Create app.py

app.py

@@ -0,0 +1,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()