app.py

# app.py
import gradio as gr
from transformers import pipeline, set_seed

generator = pipeline("text-generation", model="gpt2")
set_seed(42)

def chat_with_bot(prompt):
    response = generator(prompt, max_length=100, num_return_sequences=1)
    return response[0]["generated_text"]

iface = gr.Interface(fn=chat_with_bot, 
                     inputs="text", 
                     outputs="text", 
                     title="Mini ChatGPT",
                     description="Chatbot generatif berbasis GPT-2.")
iface.launch()