Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from transformers import pipeline,AutoConfig,AutoModelForAudioClassification
|
2 |
|
3 |
model_id = "Ahmed107/whisper-tiny-finetuned-eos"
|
4 |
|
@@ -9,7 +9,10 @@ label2id = {"NOT_EOS": "0", "EOS": "1"}
|
|
9 |
# define config
|
10 |
config = AutoConfig.from_pretrained(model_id, label2id=label2id, id2label=id2label)
|
11 |
model = AutoModelForAudioClassification.from_pretrained(model_id, config = config)
|
12 |
-
|
|
|
|
|
|
|
13 |
|
14 |
def classify_audio(filepath):
|
15 |
preds = pipe(filepath)
|
|
|
1 |
+
from transformers import pipeline,AutoConfig,AutoModelForAudioClassification,AutoFeatureExtractor
|
2 |
|
3 |
model_id = "Ahmed107/whisper-tiny-finetuned-eos"
|
4 |
|
|
|
9 |
# define config
|
10 |
config = AutoConfig.from_pretrained(model_id, label2id=label2id, id2label=id2label)
|
11 |
model = AutoModelForAudioClassification.from_pretrained(model_id, config = config)
|
12 |
+
feature_extractor = AutoFeatureExtractor.from_pretrained(
|
13 |
+
model_id,
|
14 |
+
)
|
15 |
+
pipe = pipeline("audio-classification", model=model,feature_extractor=feature_extractor)
|
16 |
|
17 |
def classify_audio(filepath):
|
18 |
preds = pipe(filepath)
|