Update app.py

app.py

@@ -10,8 +10,7 @@ translator_de = pipeline("translation", model="Helsinki-NLP/opus-mt-en-de")
 translator_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
 translator_ta = pipeline("translation", model="facebook/nllb-200-distilled-600M", src_lang="eng_Latn", tgt_lang="tam_Taml")
 speech_to_text = pipeline("automatic-speech-recognition", model="openai/whisper-small")
-question_generator = pipeline("e2e-qg")
-
+question_generator = pipeline("text2text-generation", model="valhalla/t5-base-e2e-qg")
 # Functional logic
 def summarize(text):
     return summarizer(text, max_length=60, min_length=20, do_sample=False)[0]['summary_text']
@@ -44,8 +43,10 @@ def transcribe(audio):
     return speech_to_text(audio)["text"]
 
 def generate_questions(text):
-    output = question_generator(text)
-    return "\n".join(f"- {item['question']}" for item in output[:10])
+    prompt = "generate questions: " + text
+    result = question_generator(prompt, max_length=256, do_sample=False)
+    return result[0]["generated_text"]
+
 
 # UI App
 with gr.Blocks(theme=gr.themes.Soft()) as demo: