Update app.py

app.py

@@ -48,12 +48,10 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
         input_text,
         max_new_tokens=max_new_tokens,
         temperature=temperature,
-        return_full_text = False
+        return_full_text = False,
+        forced_eos_token_id=tokenizer.encode("[...]")[1]
     )
     generated_text = output[0]['generated_text']
-    
-    if generated_text[-1] not in [".","!","?","\n"] and generated_text[-4:-1] != "</s>":
-        generated_text = generated_text + " [...]"
 
     split_tentative = generated_text.split("### Risposta:")
     if len(split_tentative) > 1: