pratikshahp commited on
Commit
7701e24
·
verified ·
1 Parent(s): 1799f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate_mcqs(text, num_questions=5):
27
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
28
  for _ in range(num_questions):
29
  # Generate a single MCQ at a time
30
- mcq = generator(text, max_length=100)[0]["generated_text"]
31
  mcqs.append(mcq)
32
 
33
  return mcqs
 
27
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
28
  for _ in range(num_questions):
29
  # Generate a single MCQ at a time
30
+ mcq = generator(text, max_length=512, max_new_tokens=100)[0]["generated_text"]
31
  mcqs.append(mcq)
32
 
33
  return mcqs