Update app.py

app.py

@@ -45,11 +45,11 @@ def invoke(
 
     tokens = tokenizer.apply_chat_template(messages, return_tensors='pt', add_generation_prompt=True)
 
-    out = model.generate(tokens, eos_token_id=model.config.eos_token_id, max_new_tokens=max_tokens)
+    out = model.generate(tokens, eos_token_id=model.config.eos_token_id, max_new_tokens=max_tokens, repetition_penalty=1.05, temperature=temperature, top_p=top_p)
 
     res = tokenizer.batch_decode(out)
 
-    return res
+    yield res
 
 
 def respond(
@@ -90,7 +90,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     invoke,
     additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        gr.Textbox(value="You are a helpful assistant.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(