Update app.py

app.py

@@ -2,15 +2,13 @@
 import gradio as gr
 from transformers import pipeline
 
-def load_question_generator():
-    pipe = pipeline("text2text-generation", model="SinaRp/Question_generator_persian", do_sample=True,
-                                top_k=50, 
-                                top_p=0.95)
+def load_question_generator(num_questions):
+    pipe = pipeline("text2text-generation", model="SinaRp/Question_generator_persian", do_sample=True, top_k=50, top_p=0.95, num_return_sequences=num_questions)
     return pipe
 
 def generate_questions(context, num_questions):
     try:
-        generator = load_question_generator()
+        generator = load_question_generator(num_questions)
         questions = generator(context, max_length=64, num_return_sequences=int(num_questions))
         return "\n\n".join([q['generated_text'] for q in questions])
     except Exception as e: