Update app.py

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