Update app.py

app.py

@@ -16,20 +16,25 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 # ======================
-# Clean output (FIX ký tự rác)
+# Clean output (fix ký tự rác)
 # ======================
-def clean_output(text: str) -> str:
-    # Cắt câu khi gặp dấu kết thúc hợp lệ
-    match = re.match(r"^[\s\S]*?[.!?\n]", text)
-    if match:
-        return match.group(0).strip()
-    return text.strip()
+def clean_output(text):
+    text = text.strip()
+    # Cắt phần dư sau câu hoàn chỉnh
+    for end in [".", "!", "?", "\n"]:
+        if end in text:
+            return text.split(end)[0] + end
+    return text
 
 # ======================
 # Chat function
 # ======================
 def chat(message, history):
-    prompt = "<bos><start_of_turn>user\n" + message + "\n<end_of_turn>\n<start_of_turn>model\n"
+    prompt = (
+        "<bos><start_of_turn>user\n"
+        + message
+        + "\n<end_of_turn>\n<start_of_turn>model\n"
+    )
 
     inputs = tokenizer(prompt, return_tensors="pt")
 
@@ -45,24 +50,23 @@ def chat(message, history):
         )
 
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
     reply = decoded.split("model")[-1].strip()
     reply = clean_output(reply)
 
     return reply
 
 # ======================
-# UI
+# UI (SAFE MODE)
 # ======================
 demo = gr.ChatInterface(
     fn=chat,
     title="🤖 Gemma3 270M Cloud Chat",
     description="Gemma3 270M chạy cloud miễn phí trên Hugging Face Spaces",
-    examples=["hi", "giải thích AI là gì", "hello world trong python"],
-    submit_btn="Send",
-    retry_btn="Retry",
-    undo_btn="Undo",
-    clear_btn="Clear"
+    examples=[
+        "hi",
+        "AI là gì?",
+        "viết hello world bằng python"
+    ]
 )
 
 if __name__ == "__main__":