Update app.py

app.py

@@ -20,11 +20,14 @@ def generate_mcqs(text, num_questions=5):
     # Tokenize the input text
     inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)
 
-    # Generate MCQs by providing the text as input to the model
-    outputs = model.generate(**inputs, max_length=100, num_return_sequences=num_questions, temperature=0.7)
-
-    # Decode the generated sequences
-    mcqs = [tokenizer.decode(output, skip_special_tokens=True) for output in outputs]
+    mcqs = []
+    for _ in range(num_questions):
+        # Generate a single MCQ at a time
+        output = model.generate(**inputs, max_length=100, temperature=0.7, num_return_sequences=1)
+
+        # Decode the generated sequence
+        mcq = tokenizer.decode(output[0], skip_special_tokens=True)
+        mcqs.append(mcq)
 
     return mcqs