Update app.py (#1)

- Update app.py (6c0b22ea68fb7dff98511ef09cd9e9268a3cfc91)


Co-authored-by: Davide Caffagni <dcaffo@users.noreply.huggingface.co>

app.py

@@ -46,7 +46,9 @@ def generate_text(prompt, slider):
     print("***** Generate *****")
     text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
     #generated_text = text_generator(prompt, max_length=100)
-    generated_text = text_generator(prompt, max_length=50, do_sample=True, temperature=slider, repetition_penalty=2.0, truncation=True)
+    if slider is not None:
+        slider = float(slider)
+    generated_text = text_generator(prompt, max_length=50, do_sample=True, temperature=slider, repetition_penalty=2.0, truncation=True, min_new_tokens=5)
     return generated_text[0]['generated_text']
 
 # Function to handle user preferences