Update app.py

app.py

@@ -52,8 +52,12 @@ def generate_text(input_text, max_new_tokens=512, temperature=1, system_prompt="
     )
     generated_text = output[0]['generated_text']
     
-    if generated_text[-1] not in [".","!","?","\n"]:
+    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:
+        generated_text = split_tentative[1]
     
     return f"<span>{input_text}</span><b style='color: blue;'>{generated_text}</b>"