Update app.py

app.py

@@ -13,13 +13,22 @@ def get_question(context, answer):
     #encoding = question_tokenizer.encode_plus(text, max_length=max_len, padding='max_length', truncation=True, return_tensors="pt")
     encoding = question_tokenizer.encode_plus(text, return_tensors="pt")
     input_ids, attention_mask = encoding["input_ids"], encoding["attention_mask"]
+    #outs = question_model.generate(input_ids=input_ids,
+    #                                attention_mask=attention_mask,
+    #                                early_stopping=True,
+    #                                num_beams=3, # Use fewer beams to generate fewer but higher-quality questions
+    #                                num_return_sequences=3,
+    #                                no_repeat_ngram_size=3, # Allow some repetition to avoid generating nonsensical questions
+    #                                max_length=256) # Use a shorter max length to focus on generating more relevant questions
+
     outs = question_model.generate(input_ids=input_ids,
-                                    attention_mask=attention_mask,
-                                    early_stopping=True,
-                                    num_beams=3, # Use fewer beams to generate fewer but higher-quality questions
-                                    num_return_sequences=3,
-                                    no_repeat_ngram_size=3, # Allow some repetition to avoid generating nonsensical questions
-                                    max_length=256) # Use a shorter max length to focus on generating more relevant questions
+                                attention_mask=attention_mask,
+                                early_stopping=True,)
+
+
+
+
+    
     dec = [question_tokenizer.decode(ids) for ids in outs]
     questions = ""
     for i, question in enumerate(dec):