import spaces
from transformers import pipeline

model = pipeline("text-generation", model="gpt2")

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

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