Update app.py

app.py

@@ -1,8 +1,14 @@
-import gradio as gr
+from transformers import pipeline
 
-model_id = "sanchit-gandhi/whisper-small-dv"  # update with your model id
+model_id = "Teapack1/model_KWS"  # update with your model id
 pipe = pipeline("automatic-speech-recognition", model=model_id)
 
+import gradio as gr
+
+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 transcribe_speech(filepath):
     output = pipe(
         filepath,