Update app.py

app.py

@@ -1,3 +1,15 @@
-import gradio as gr
+import spaces
+from diffusers import DiffusionPipeline
 
-gr.load("models/ptx0/terminus-xl-velocity-v2").launch()
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
+
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()