Update app.py

app.py

@@ -72,11 +72,12 @@ def generate(
         {"input_ids": input_ids},
         streamer=streamer,
         max_new_tokens=max_new_tokens,
-        do_sample=False if (temperature == 0 or top_k == 1) else True,
+        do_sample=False if top_k == 1 else True,
         top_p=top_p,
         top_k=top_k,
         temperature=temperature,
         num_beams=1,
+        repetition_penalty=1.0,
     )
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
@@ -105,16 +106,16 @@ chat_interface = gr.ChatInterface(
         gr.Slider(
             label="Temperature",
             minimum=0.1,
-            maximum=4.0,
+            maximum=2.0,
             step=0.1,
-            value=1.0,
+            value=0.7,
         ),
         gr.Slider(
             label="Top-p (nucleus sampling)",
             minimum=0.05,
             maximum=1.0,
             step=0.05,
-            value=1.0,
+            value=0.9,
         ),
         gr.Slider(
             label="Top-k",