ahmad-fakhar commited on
Commit
800a812
·
verified ·
1 Parent(s): be38906

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import os
6
  st.title('Music Genre Classification')
7
 
8
  # Load the model and feature extractor
9
- model = TFAutoModelForAudioClassification.from_pretrained("your-tensorflow-compatible-model")
10
- feature_extractor = AutoFeatureExtractor.from_pretrained("your-tensorflow-compatible-model")
11
 
12
  # Create the audio classification pipeline
13
  pipe = pipeline(task="audio-classification", model=model, feature_extractor=feature_extractor)
 
6
  st.title('Music Genre Classification')
7
 
8
  # Load the model and feature extractor
9
+ model = TFAutoModelForAudioClassification.from_pretrained("sandychoii/distilhubert-finetuned-gtzan-audio-classificationyour-tensorflow-compatible-model")
10
+ feature_extractor = AutoFeatureExtractor.from_pretrained("sandychoii/distilhubert-finetuned-gtzan-audio-classification")
11
 
12
  # Create the audio classification pipeline
13
  pipe = pipeline(task="audio-classification", model=model, feature_extractor=feature_extractor)