Update app.py

app.py

@@ -12,7 +12,7 @@ pipe = pipeline(
 
 @spaces.GPU
 def infer(prompt):
-    return pipe(prompt, max_new_tokens=1, do_sample=True, temperature=1.0)[0]["generated_text"]
+    return pipe(prompt, max_new_tokens=1, do_sample=True, temperature=1.0, return_full_text=False)[0]["generated_text"]
 
 with gr.Blocks(fill_width=True, css="""
 #prompt-box textarea {height:200px}