Update app.py

app.py

@@ -15,14 +15,18 @@ headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 # ----------------- FEATURE FUNCTIONS -----------------
 
 def generate_quiz(text, num_questions):
-    API_URL = "https://api-inference.huggingface.co/models/mrm8488/t5-base-finetuned-question-generation-ap"
-    payload = {"inputs": f"generate questions: {text}", "parameters": {"max_length": 256}}
+    prompt = f"Generate {num_questions} MCQ questions from this content:\n{text}"
+    API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
+    payload = {
+        "inputs": prompt,
+        "parameters": {"max_new_tokens": 256}
+    }
     response = requests.post(API_URL, headers=headers, json=payload)
     try:
-        result = response.json()[0]["generated_text"]
-        return f"{num_questions} Sample Questions:\n\n{result}"
-    except:
-        return "⚠️ Failed to generate quiz. Try again later."
+        return response.json()[0]["generated_text"]
+    except Exception as e:
+        return f"⚠️ Error: {e}"
+
 
 def get_bot_response(query):
     API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"