make diversity_penalty float

app.py

@@ -81,7 +81,7 @@ def suggest_next(text, model_name, k, m, num_beam_groups, diversity_penalty):
     }
     if diversity_penalty and diversity_penalty > 0:
         gen_kwargs["num_beam_groups"] = num_beam_groups
-        gen_kwargs["diversity_penalty"] = diversity_penalty
+        gen_kwargs["diversity_penalty"] = float(diversity_penalty)
 
     outs = gen_pipe(text, **gen_kwargs)