import spaces
from transformers import pipeline

# Load your model
model = pipeline("text-generation", model="gpt2")
model.to("cuda")

@spaces.GPU
def generate_text(prompt):
 return model(prompt)

import gradio as gr
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
iface.launch()