fix num beams

app.py

@@ -16,8 +16,10 @@ def get_answer(user_input, decoding_methods, num_beams, top_k, top_p, temperatur
     elif decoding_methods == "Sampling":
         do_sample = True
         penalty_alpha = 0
+        num_beams = 1
     else:
         do_sample = False
+        num_beams = 1
     print(user_input, decoding_methods, do_sample, top_k, top_p, temperature, repetition_penalty, penalty_alpha)
     prompt = f"User: {user_input}\nAssistant: "
     generated_text = text_generation(f"{prompt}", min_length=50, max_length=200, num_return_sequences=1,
@@ -46,7 +48,8 @@ with gr.Blocks() as demo:
             top_p = gr.inputs.Slider(label="Top P", default=0.9, step=0.05, minimum=0.1, maximum=1.0)
             temperature = gr.inputs.Slider(label="Temperature", default=0.5, step=0.05, minimum=0.1, maximum=1.0)
             repetition_penalty = gr.inputs.Slider(label="Repetition Penalty", default=1.1, step=0.05, minimum=1.0, maximum=2.0)
-            penalty_alpha = gr.inputs.Slider(label="The penalty alpha for contrastive search", default=1.1, step=0.05, minimum=1.0, maximum=2.0)
+            penalty_alpha = gr.inputs.Slider(label="The penalty alpha for contrastive search",
+                                             default=0.5, step=0.05, minimum=0.05, maximum=1.0)
             with gr.Row():
                 button_generate_story = gr.Button("Submit")
         with gr.Column():