Update app.py (#10)

- Update app.py (7087540406264c37a0959b46b413773a7970fd35)

app.py

@@ -6,9 +6,9 @@ default_lang = "en"
 
 # ChatGPT
 from pyChatGPT import ChatGPT
-#import whisper
-#whisper_model = whisper.load_model("small")
-whisper = gr.Blocks.load(name="spaces/sanchit-gandhi/whisper-large-v2")
+import whisper
+whisper_model = whisper.load_model("small")
+#whisper = gr.Blocks.load(name="spaces/sanchit-gandhi/whisper-large-v2")
 import os
 session_token = os.environ.get('SessionToken') 
 
@@ -46,30 +46,6 @@ def chat_hf(audio, custom_token, language):
     return whisper_text, gpt_response, fp.name
 
 # whisper
-#def translate(audio):
-#    print("""
-#    —
-#    Sending audio to Whisper ...
-#    —
-#    """)
-#    
-#    audio = whisper.load_audio(audio)
-#    audio = whisper.pad_or_trim(audio)
-#    
-#    mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
-#    
-#    _, probs = whisper_model.detect_language(mel)
-#    
-#    transcript_options = whisper.DecodingOptions(task="transcribe", fp16 = False)
-#    
-#    transcription = whisper.decode(whisper_model, mel, transcript_options)
-#    
-#    print("language spoken: " + transcription.language)
-#    print("transcript: " + transcription.text)
-#    print("———————————————————————————————————————————")  
-#      
-#    return transcription.text
-
 def translate(audio):
     print("""
     —
@@ -77,9 +53,33 @@ def translate(audio):
     —
     """)
     
-    text_result = whisper(audio, None, "transcribe", fn_index=0)
-    print(text_result)
-    return text_result
+    audio = whisper.load_audio(audio)
+    audio = whisper.pad_or_trim(audio)
+    
+    mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
+    
+    _, probs = whisper_model.detect_language(mel)
+    
+    transcript_options = whisper.DecodingOptions(task="transcribe", fp16 = False)
+    
+    transcription = whisper.decode(whisper_model, mel, transcript_options)
+    
+    print("language spoken: " + transcription.language)
+    print("transcript: " + transcription.text)
+    print("———————————————————————————————————————————")  
+      
+    return transcription.text
+
+#def translate(audio):
+#    print("""
+#    —
+#    Sending audio to Whisper ...
+#    —
+#    """)
+#   
+#    text_result = whisper(audio, None, "transcribe", fn_index=0)
+#    print(text_result)
+#    return text_result
 
 
 with gr.Blocks() as blocks: