Update app.py

app.py

@@ -2,10 +2,12 @@ import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 import tempfile
+import pygame
+import time
 
 # Load the question-answering model
 model_name = "AVISHKAARAM/avishkaarak-ekta-hindi"
-model = pipeline("question-answering", model=model_name, tokenizer=model_name)
+model = pipeline("question-answering", model=model_name)
 
 def question_answering(context, question):
     # Use the question-answering model to get the answer
@@ -19,12 +21,21 @@ def question_answering(context, question):
 
     return answer, audio_path
 
+def play_audio(audio_path):
+    pygame.mixer.init()
+    pygame.mixer.music.load(audio_path)
+    pygame.mixer.music.play()
+    while pygame.mixer.music.get_busy():
+        time.sleep(0.1)
+
 # Create the Gradio interface
 iface = gr.Interface(
     fn=question_answering,
     inputs=["text", "text"],
-    outputs=[gr.outputs.Textbox(label="Answer"), gr.outputs.File(label="Answer Audio")],
-    layout="vertical",
+    outputs=[
+        gr.outputs.Textbox(label="Answer"),
+        gr.outputs.Audio(label="Answer Audio", type="numpy")
+    ],
     title="Question Answering - AVISHKAARAK",
     description="Enter a context and a question to get an answer. :)",
     examples=[
@@ -34,5 +45,4 @@ iface = gr.Interface(
 )
 
 # Run the interface
-iface.launch()
-
+iface.launch()