Update app.py

app.py

@@ -428,7 +428,7 @@ def get_beam_search_html(
     outputs = model.generate(
         **inputs,
         max_new_tokens=number_steps,
-        num_beams=max(number_beams, 2),
+        num_beams=number_beams,
         num_return_sequences=num_return_sequences,
         return_dict_in_generate=True,
         length_penalty=length_penalty,