Update app.py

app.py

@@ -1,50 +1,53 @@
 import gradio as gr
-from transformers import pipeline, set_seed
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-# Load model
-text_generator = pipeline("text-generation", model="NlpHUST/gpt2-vietnamese", tokenizer="NlpHUST/gpt2-vietnamese")
-set_seed(42)
+# Tải model và tokenizer
+model_name = "NlpHUST/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 = prompt.strip()
-    if not prompt:
-        return "❗️Vui lòng nhập thông tin mô tả bản thân để tạo bio."
-
-    full_prompt = f"Viết một đoạn bio ngắn, giới thiệu bản thân với nội dung:\n{prompt}\n\nBio:"
-
-    try:
-        outputs = text_generator(
-            full_prompt,
-            max_length=min(max_length, 80),
-            temperature=temperature,
-            do_sample=True,
-            top_k=50,
-            top_p=0.95,
-            num_return_sequences=1,
-            eos_token_id=text_generator.tokenizer.eos_token_id
-        )
-        result = outputs[0]["generated_text"].replace(full_prompt, "").strip()
-        if len(result) > max_length:
-            result = result[:max_length].rsplit('.', 1)[0] + '.'
-        return result if result else "⚠️ Không tạo được bio phù hợp. Hãy thử lại với mô tả khác."
-    except Exception as e:
-        return f"🚫 Lỗi xảy ra khi sinh văn bản: {str(e)}"
+    # 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:"
+    )
+    
+    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
+    )
+    
+    # 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
 
-# Tạo Gradio app
-with gr.Blocks() as app:
+# 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:
     gr.Markdown("✨ **AI Viết Bio Thương Hiệu Cá Nhân (Tiếng Việt)**")
 
     with gr.Row():
         with gr.Column():
-            input_text = gr.Textbox(label="🧠 Mô tả bản thân / điểm mạnh")
-            max_len = gr.Slider(20, 300, value=60, label="📏 Độ dài tối đa")
-            temp = gr.Slider(0.1, 1.0, value=0.8, label="🌡️ Mức sáng tạo (temperature)")
+            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")
+
         with gr.Column():
-            output = gr.Textbox(label="✅ Đoạn bio gợi ý", lines=6)
+            output_box = gr.Textbox(label="📋 Đoạn bio gợi ý", lines=5)
     
-    submit_btn.click(fn=generate_bio, inputs=[input_text, max_len, temp], outputs=output)
+    # 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)
 
-# 🚀 Quan trọng: Lệnh launch app
 app.launch()