Update app.py

app.py

@@ -1,3 +1,11 @@
+import gradio as gr
+from transformers import pipeline, set_seed
+
+# Load model
+text_generator = pipeline("text-generation", model="NlpHUST/gpt2-vietnamese", tokenizer="NlpHUST/gpt2-vietnamese")
+set_seed(42)
+
+# Hàm sinh bio
 def generate_bio(prompt, max_length, temperature):
     prompt = prompt.strip()
     if not prompt:
@@ -8,22 +16,35 @@ def generate_bio(prompt, max_length, temperature):
     try:
         outputs = text_generator(
             full_prompt,
-            max_length=min(max_length, 80),  # Giới hạn cứng tránh vượt quá
+            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  # Dừng đúng chỗ
+            eos_token_id=text_generator.tokenizer.eos_token_id
         )
-        # Cắt kết quả đúng phần sinh ra
         result = outputs[0]["generated_text"].replace(full_prompt, "").strip()
-        # Giới hạn độ dài đầu ra
         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)}"
 
+# Tạo Gradio 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():
+            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)")
+            submit_btn = gr.Button("Submit")
+        with gr.Column():
+            output = gr.Textbox(label="✅ Đoạn bio gợi ý", lines=6)
+    
+    submit_btn.click(fn=generate_bio, inputs=[input_text, max_len, temp], outputs=output)
 
+# 🚀 Quan trọng: Lệnh launch app
+app.launch()