Update app.py

app.py

@@ -20,6 +20,8 @@ def generate(
     pipe = load_model(model_name)
 
     # Set tokenize correctly. Otherwise ticking the box breaks it.
+    if repetition_penalty == 0:
+        repetition_penalty = 1
     if model_name == "M4-ai/tau-1.8B":
         prompt = user_input
     else: