Update app.py

app.py

@@ -6,16 +6,11 @@ def load_question_generator():
     pipe = pipeline("text2text-generation", model="SinaRp/Question_generator_persian")
     return pipe
 
-def generate_questions(context, num_questions=3):
+def generate_questions(context, num_questions):
     try:
         generator = load_question_generator()
-        questions = []
-        
-        for _ in range(num_questions):
-            result = generator(context, max_length=64, num_return_sequences=1)
-            questions.append(result[0]['generated_text'])
-            
-        return "\n\n".join(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:
         return f"Error generating questions: {str(e)}"
 
@@ -30,8 +25,9 @@ iface = gr.Interface(
     description="Generate questions from your text using AI",
     examples=[
         ["The Sun is the star at the center of the Solar System. It is a nearly perfect sphere of hot plasma, heated to incandescence by nuclear fusion reactions in its core.", 2],
+        ["Machine learning is a subset of artificial intelligence that focuses on the use of data and algorithms to imitate the way that humans learn.", 3]
     ]
 )
 
-iface.launch()
-
+if __name__ == "__main__":
+    iface.launch()