File size: 531 Bytes
3b0d46a 49a0464 3b0d46a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | 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()
|