fix translation

app.py

@@ -30,6 +30,7 @@ def main ():
         st.audio(wav_audio_data, format='audio/wav')
         original = transcribe(wav_audio_data, s2t)
         st.write(f"Original: {original}")
+        print("next is translation")
         print("translating audio...")
         translation = translate(original)
         st.write(f"Transcription: {translation}")

speech2text.py

@@ -33,7 +33,7 @@ def init_model_trans ():
         device=device,
     )
     print(f'Init model successful')
-    audit_elapsedtime(function="Initiating transcription model", start=start)
+
     return pipe
 
 def transcribe (audio_sample: bytes, pipe) -> str:
@@ -51,17 +51,17 @@ def transcribe (audio_sample: bytes, pipe) -> str:
     #st.write('trancription: ', result["text"])
     return result["text"]
 
-def translate (audio_sample: bytes, pipe) -> str:
-    print("Initiating Translation...")
-    start = set_start()
-    # dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
-    # sample = dataset[0]["audio"]
+# def translate (audio_sample: bytes, pipe) -> str:
+#     print("Initiating Translation...")
+#     start = set_start()
+#     # dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")
+#     # sample = dataset[0]["audio"]
     
-    #result = pipe(audio_sample)
-    result = pipe(audio_sample, generate_kwargs={"task": "translate"})
+#     #result = pipe(audio_sample)
+#     result = pipe(audio_sample, generate_kwargs={"task": "translate"})
 
-    audit_elapsedtime(function="Translation", start=start)
-    print("Translation result",result)
+#     audit_elapsedtime(function="Translation", start=start)
+#     print("Translation result",result)
 
-    #st.write('trancription: ', result["text"])
-    return result["text"]
+#     #st.write('trancription: ', result["text"])
+#     return result["text"]