from transformers import pipeline
import gradio as gr


model_name = "HuyRemy/Ai-bocphet"
chatbot = pipeline("text-generation", model=model_name)

def generate_response(prompt):
    response = chatbot(prompt, max_length=150, num_return_sequences=1)
    return response[0]['generated_text']

# Ví dụ sử dụng
iface = gr.Interface(
    fn=generate_response,
    inputs="text",
    outputs="text",
    title="AI-Bốc Phét (sáng tác truyện)",
    description="Example: Extend this story. HuyRemy is a super hero in Vietnam, he's like Iron Man from USA",
)

# Bắt đầu ứng dụng
iface.launch()