Update app.py

app.py

@@ -124,6 +124,9 @@ def main():
     length_penalty = st.sidebar.number_input(
         "Length penalty", min_value=0.0, max_value=2.0, value=1.2, step=0.1
     )
+    diversity_penalty = st.sidebar.number_input(
+        "Diversity penalty", min_value=0.0, max_value=2.0, value=0.0, step=0.1
+    )
     st.sidebar.markdown(
         """For an explanation of the parameters, head over to the [Huggingface blog post about text generation](https://huggingface.co/blog/how-to-generate)
 and the [Huggingface text generation interface doc](https://huggingface.co/transformers/main_classes/model.html?highlight=generate#transformers.generation_utils.GenerationMixin.generate).
@@ -132,6 +135,7 @@ and the [Huggingface text generation interface doc](https://huggingface.co/trans
     params = {
         "num_beams": num_beams,
         "num_beam_groups": num_beam_groups,
+        "diversity_penalty": diversity_penalty if num_beam_groups > 1 else 0.0,
         "length_penalty": length_penalty,
         "early_stopping": True,
     }