Upgrade

app.py

@@ -1,23 +1,21 @@
-import os
-os.system("pip install git+https://github.com/openai/whisper.git")
 import gradio as gr
-import whisper
 
-model = whisper.load_model("small")
+whisper = gr.load("models/openai/whisper-small")
         
 def inference(audio):
-    audio = whisper.load_audio(audio)
-    audio = whisper.pad_or_trim(audio)
+    # audio = whisper.load_audio(audio)
+    # audio = whisper.pad_or_trim(audio)
     
-    mel = whisper.log_mel_spectrogram(audio).to(model.device)
+    # mel = whisper.log_mel_spectrogram(audio).to(model.device)
     
-    _, probs = model.detect_language(mel)
+    # _, probs = model.detect_language(mel)
     
-    options = whisper.DecodingOptions(fp16 = False)
-    result = whisper.decode(model, mel, options)
+    # options = whisper.DecodingOptions(fp16 = False)
+    # result = whisper.decode(model, mel, options)
     
-    print(result.text)
-    return result.text
+    # print(result.text)
+    # return result.text
+    return whisper(audio).replace("AutomaticSpeechRecognitionOutput(text=' ", "").replace("', chunks=None)", "")
 
 article = """
 ## 参考