Update app.py

app.py

@@ -62,17 +62,18 @@ def get_question(sentence,answer):
   outs = question_model.generate(input_ids=input_ids,
                                   attention_mask=attention_mask,
                                   early_stopping=True,
-                                  num_beams=5,
-                                  num_return_sequences=1,
+                                  num_beams=3,
+                                  num_return_sequences=3,
                                   no_repeat_ngram_size=2,
                                   max_length=200)
 
 
   dec = [question_tokenizer.decode(ids) for ids in outs]
 
-  Question = dec[0].replace("question:","")
-  Question= Question.strip()
-  return Question
+  #Question = dec[0].replace("question:","")
+  #Question= Question.strip()
+  #return Question
+  return dec
 
 input_context = gr.Textbox()
 input_answer = gr.Textbox()