File size: 252 Bytes
5edc4cb
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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()