Add application file

app.py

@@ -39,10 +39,10 @@ def get_answer_from_document(question, context):
     result = qa_pipeline(question=question, context=context)
     return result
 
-def generate_questions(text, num_questions=5):
+def generate_questions(text, num_questions=5, num_beams=5):
     question_generation_pipeline = pipeline("text2text-generation", model="valhalla/t5-base-qg-hl")
     input_text = "generate questions: " + text
-    questions = question_generation_pipeline(input_text, max_length=512, num_return_sequences=num_questions)
+    questions = question_generation_pipeline(input_text, max_length=512, num_beams=num_beams, num_return_sequences=num_questions)
     return [q['generated_text'] for q in questions]
 
 def evaluate_responses(user_responses, correct_answers):
@@ -91,3 +91,4 @@ if uploaded_file is not None:
         training_plan = generate_training_plan(scores)
         for step in training_plan:
             st.write(step)
+