Update app.py

app.py

@@ -35,10 +35,10 @@ def genQuestion(model_choice, context):
         # descriptions = "Reverse interview question  is a model that..."
 
     inputs = tok(context, return_tensors="pt")
-    output = model.generate(inputs["input_ids"], num_beams=16, max_length=64, min_length=9, num_return_sequences=16, diversity_penalty = 1.0, num_beam_groups=4)
+    output = model.generate(inputs["input_ids"], num_beams=4, max_length=64, min_length=9, num_return_sequences=4, diversity_penalty = 1.0, num_beam_groups=4)
     final_output = ''
     print(output)
-    for i in range(16):
+    for i in range(4):
         final_output +=  [tok.decode(beam, skip_special_tokens=True, clean_up_tokenization_spaces=False) for beam in output][i] + "\n\n"
 
     return final_output