Update app.py
Browse files
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("
|
10 |
-
feature_extractor = AutoFeatureExtractor.from_pretrained("
|
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)
|