Update app

app.py

@@ -1,13 +1,19 @@
 from transformers import pipeline
 import gradio as gr
+from transformers import WhisperForConditionalGeneration
 
 pipe = pipeline(model="matteocirca/whisper-small-it")
 # pipe = pipeline(model="openai/whisper-small")
+# model = WhisperForConditionalGeneration.from_pretrained("matteocirca/whisper-small-it")
+
 segments = []
+
 def audio2segments(audio,word):
     global segments
     if not segments:
-        segments = pipe(audio)["segments"]
+        print(pipe(audio).keys())
+        # segments = pipe(audio)["segments"]
+        # segments = model.transcribe(audio)["segments"]
     elif not word:
         return "No word detected"
     else: