Update app.py

app.py

@@ -375,7 +375,7 @@ def generate_beams(start_sentence, scores, length_penalty, decoded_sequences):
                     step == len(scores) - 1
                     or current_token_choice_ix == tokenizer.eos_token_id
                 ),
-                is_selected_sequence=(current_sequence in decoded_sequences),
+                is_selected_sequence=(current_sequence.replace('<|endoftext|>', '') in [el.replace('<|endoftext|>', '') for el in decoded_sequences]),
             )
 
         # Reassign all beams at once
@@ -465,7 +465,7 @@ This parameter will not impact the beam search paths, but only influence the cho
             label="Length penalty", minimum=-3, maximum=3, step=0.5, value=1
         )
         num_return_sequences = gr.Slider(
-            label="Number of return sequences", minimum=1, maximum=4, step=1, value=3
+            label="Number of return sequences", minimum=1, maximum=3, step=1, value=2
         )
 
     n_beams.change(fn=change_num_return_sequences, inputs=n_beams, outputs=num_return_sequences)