import gradio as gr
from transformers import pipeline

# Load the text generation model
generator = pipeline("text-generation", model="gpt2")

# Define a function for the API
def generate_text(prompt):
    result = generator(prompt, max_length=100, num_return_sequences=1)
    return result[0]["generated_text"]

# Set up a Gradio interface
interface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Chat Gpt 2",
    description="wowo its mr gpt but 2"
)

# Launch the app
interface.launch()