Update app.py

app.py

@@ -44,7 +44,7 @@ if text != '':
             top_k=50,
             top_p=0.95,
             early_stopping=True,
-            num_return_sequences=5
+            num_return_sequences=10
         )
     elif str(decoding_strategy) == 'Beam Search':
         beam_outputs = model.generate(
@@ -53,7 +53,7 @@ if text != '':
         max_length=20, 
         num_beams=5, 
         no_repeat_ngram_size=2, 
-        num_return_sequences=5, 
+        num_return_sequences=10, 
         early_stopping=True
         )
         
@@ -62,6 +62,8 @@ if text != '':
         sent = tokenizer.decode(beam_output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
         if sent.lower() != text.lower() and sent not in final_outputs:
             final_outputs.append(sent)
+        if len(final_outputs) == 5:
+            break
         # final_outputs.append(sent)
 
     st.subheader('Paraphrased Text')