Update app.py

app.py

@@ -1,7 +1,15 @@
-from transformers import pipeline
+from transformers import pipeline,AutoConfig,AutoModelForAudioClassification
 
 model_id = "Ahmed107/whisper-tiny-finetuned-eos"
-pipe = pipeline("audio-classification", model=model_id)
+
+# define mappings as dictionaries
+id2label = {"0": "NOT_EOS", "1": "EOS"}
+label2id = {"NOT_EOS": "0", "EOS": "1"}
+
+# define config
+config = AutoConfig.from_pretrained(model_id, label2id=label2id, id2label=id2label)
+model = AutoModelForAudioClassification.from_pretrained(model_id, config = config)
+pipe = pipeline("audio-classification", model=model)
 
 def classify_audio(filepath):
     preds = pipe(filepath)