Update app.py

app.py

@@ -21,12 +21,18 @@ def generate_mcqs(text, num_questions=5):
     model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-large")
 
     # Tokenize the input text
-    inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)
+    inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
 
     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)
+        output = model.generate(
+            **inputs,
+            max_length=512,  # Adjusted to match the tokenized input length
+            max_new_tokens=100,  # Limit the number of new tokens added during generation
+            temperature=0.7,
+            num_return_sequences=1
+        )
 
         # Decode the generated sequence
         mcq = tokenizer.decode(output[0], skip_special_tokens=True)