Update app.py

app.py

@@ -3,11 +3,11 @@ import gradio as gr
 from transformers import pipeline
 
 # ====== Load AI models ======
-# Summarization model
+# Summarization model (BART)
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Question generation model (for quiz & flashcards)
-question_generator = pipeline("text2text-generation", model="iarfmoose/t5-base-question-generator")
+# Question generation model (T5 small, CPU-friendly)
+question_generator = pipeline("text2text-generation", model="valhalla/t5-small-qg-hl")
 
 # ====== Summarizer Function ======
 def summarize_text(text, mode):
@@ -29,7 +29,7 @@ def generate_quiz(text):
     if not text.strip():
         return "⚠️ Please provide some notes!"
     # Generate 5 questions
-    output = question_generator(f"Generate 5 quiz questions from the following text:\n{text}", max_length=256)
+    output = question_generator(f"generate questions: {text}", max_length=256)
     return output[0]['generated_text']
 
 # ====== Flashcards Generator ======
@@ -37,7 +37,7 @@ def generate_flashcards(text):
     if not text.strip():
         return "⚠️ Please provide some notes!"
     # Generate Q&A style flashcards
-    output = question_generator(f"Generate flashcards (Q&A) from the following text:\n{text}", max_length=256)
+    output = question_generator(f"generate flashcards: {text}", max_length=256)
     return output[0]['generated_text']
 
 # ====== Gradio Interface ======
@@ -61,4 +61,3 @@ with gr.Blocks() as demo:
         gr.Button("Generate Flashcards").click(fn=generate_flashcards, inputs=[flash_input], outputs=[flash_output])
 
 demo.launch()
-