Update app.py

app.py

@@ -9,30 +9,18 @@ pipe = pipeline("audio-classification", model=model_id)
 title = "Keyword Spotting Wav2Vec2"
 description = "Gradio demo for finetuned Wav2Vec2 model on a custom dataset to perform keyword spotting task. Classes are scene 1, scene 2, scene 3, yes, no and stop."
 
-def classify_audio(filepath):
-    preds = pipe(filepath,
-                 chunk_length_s=1,
-                 )
-    outputs = {}
-    for p in preds:
-        outputs[p["label"]] = p["score"]
-    return outputs
 
+def classify_audio(audio):
+    predictions = audio_classifier(audio)
+    return predictions
+    
 
-demo = gr.Blocks()
-
-mic_classify = gr.Interface(
-    title = title,
-    description=description,
+demo = gr.Interface(
     fn=classify_audio,
-    inputs=gr.Audio(sources="microphone", type="filepath"),
-    outputs=gr.outputs.Label()
+    inputs=gr.inputs.Audio(source="microphone", type="numpy", label="Record your audio"),
+    outputs="label",
+    title="Audio Classification Demo",
+    description="A simple demo to classify audio using a Hugging Face model."
 )
 
-
-with demo:
-    gr.TabbedInterface(
-        [mic_classify, file_classify],
-        ["Transcribe Microphone", "Transcribe Audio File"],
-    )
 demo.launch(debug=True, share=True)