Update app.py

app.py

@@ -133,11 +133,11 @@ if  num in [str(x) for x in list(range(10))]:
                             min_new_tokens = tokens,
                             top_p = top_p,
                             top_k = top_k,
-                            num_return_sequences = 5,
+                            num_return_sequences = beams,
                             )
             output_list = []
             # st.write(input)
-            for x in range(5):
+            for x in range(beams):
                 output_text = tokenizer.decode(output_ids[x], skip_special_tokens=True,
                                                     clean_up_tokenization_spaces=False)
                 output_list.append(output_text)