Update app.py

app.py

@@ -1,53 +1,47 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-# Tải model và tokenizer
-model_name = "NlpHUST/gpt2-vietnamese"
+model_name = "antphb/gpt2-vietnamese"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Hàm sinh bio
 def generate_bio(prompt, max_length, temperature):
-    # Prompt cứng để định hướng model
     full_prompt = (
-        f"Viết một đoạn bio 1-2 câu, ngắn gọn, vui nhộn, dùng trên mạng xã hội.\n"
-        f"Chủ đề: {prompt.strip()}\nBio:"
+        f"Viết 1 đoạn bio 1-2 câu, ngắn gọn, vui nhộn, dùng trên mạng xã hội.\n"
+        f"Chủ đề: {prompt.strip()}.\nBio:"
     )
-    
     outputs = text_generator(
         full_prompt,
         max_length=max_length,
         do_sample=True,
         temperature=temperature,
         pad_token_id=tokenizer.eos_token_id,
-        num_return_sequences=1,
         top_k=50,
-        top_p=0.95
+        top_p=0.95,
+        num_return_sequences=1
     )
-    
-    # Xử lý đầu ra, loại bỏ phần prompt ban đầu
     generated = outputs[0]["generated_text"]
-    generated_bio = generated.replace(full_prompt, "").strip()
-    return generated_bio
+    bio = generated.replace(full_prompt, "").strip()
+    sentences = bio.split(".")
+    result = ". ".join(sentences[:2]).strip()
+    if not result.endswith("."):
+        result += "."
+    return result
 
-# Tạo giao diện
-with gr.Blocks(title="AI Viết Bio Thương Hiệu Cá Nhân (Tiếng Việt)") as app:
+with gr.Blocks() as app:
     gr.Markdown("✨ **AI Viết Bio Thương Hiệu Cá Nhân (Tiếng Việt)**")
-
     with gr.Row():
         with gr.Column():
-            prompt_input = gr.Textbox(label="🧠 Mô tả bản thân / điểm mạnh", placeholder="VD: Tôi là người chơi cầu lông hài hước")
-            max_length = gr.Slider(label="✏️ Độ dài tối đa", minimum=20, maximum=300, value=60, step=5)
-            temperature = gr.Slider(label="🧪 Mức sáng tạo (temperature)", minimum=0.1, maximum=1.0, value=0.8, step=0.1)
-            submit_btn = gr.Button("Submit")
-            clear_btn = gr.Button("Clear")
-
+            prompt_input = gr.Textbox(lines=2, label="Mô tả bản thân / điểm mạnh")
+            max_length = gr.Slider(20, 200, value=60, step=5, label="Độ dài tối đa")
+            temperature = gr.Slider(0.1, 1.0, value=0.8, step=0.1, label="Mức sáng tạo")
+            submit_btn = gr.Button("Tạo bio")
+            clear_btn = gr.Button("Xóa")
         with gr.Column():
-            output_box = gr.Textbox(label="📋 Đoạn bio gợi ý", lines=5)
-    
-    # Xử lý nút
-    submit_btn.click(fn=generate_bio, inputs=[prompt_input, max_length, temperature], outputs=output_box)
-    clear_btn.click(fn=lambda: "", inputs=None, outputs=output_box)
+            output_box = gr.Textbox(lines=4, label="Bio gợi ý")
+    submit_btn.click(generate_bio, [prompt_input, max_length, temperature], output_box)
+    clear_btn.click(lambda: "", None, output_box)
 
 app.launch()
+