Update app.py

app.py

@@ -11,17 +11,14 @@ pipe = pipeline(
 
 def chat_fn(message, history):
     history = history or []
-    prompt = message
-
     outputs = pipe(
-        prompt,
+        message,
         max_new_tokens=256,
         do_sample=True,
         temperature=0.7,
+        return_full_text=False,
     )
-
-    text = outputs[0]["generated_text"]
-    return text
+    return outputs[0]["generated_text"]
 
 demo = gr.ChatInterface(fn=chat_fn)