Update app.py

app.py

@@ -18,10 +18,11 @@ pipe = pipeline(
     "text-generation",
     model=model_name,
     tokenizer=model_name,
-    max_new_tokens=400,
+    max_new_tokens=250,
     temperature=0.5,
     num_beams=6,
-    repetition_penalty=1.5
+    repetition_penalty=1.5,
+    return_full_text=False
 )
 
 # Prompt template