Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import whisper
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from gtts import gTTS
-from IPython.display import Audio
+import base64
 
 # Load the models and tokenizer
 whisper_model = whisper.load_model("base")
@@ -40,7 +40,10 @@ def translate_audio(input_file, to_lang):
     # Load audio data from file
     audio_data = open(temp_output_file, "rb").read()
     
-    return Audio(audio_data)
+    # Convert audio data to Base64 string
+    audio_base64 = base64.b64encode(audio_data).decode("utf-8")
+    
+    return audio_base64
 
 def translate_audio_interface(input_file, to_lang):
     return translate_audio(input_file, to_lang)
@@ -48,7 +51,7 @@ def translate_audio_interface(input_file, to_lang):
 iface = gr.Interface(
     fn=translate_audio_interface,
     inputs=["file", "text"],
-    outputs="audio",
+    outputs="text",
     title="Audio Translation",
     description="Upload an MP3 file and select the target language for translation.",
     examples=[
@@ -57,4 +60,5 @@ iface = gr.Interface(
     ]
 )
 
-iface.launch(debug = True)
+iface.launch()
+