Update app.py

app.py

@@ -1,26 +1,26 @@
-import gradio as gr
-import spaces
-import torch
+# import gradio as gr
+# import spaces
+# import torch
 
-@spaces.GPU
-def greet(name):
-    return f"Tensor: {torch.Tensor([0]).cuda()}, Name: {name}"
+# @spaces.GPU
+# def greet(name):
+#     return f"Tensor: {torch.Tensor([0]).cuda()}, Name: {name}"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+# iface.launch()
 
-# import spaces
-# from diffusers import DiffusionPipeline
+import spaces
+from diffusers import DiffusionPipeline
 
-# pipe = DiffusionPipeline.from_pretrained(...)
-# pipe.to('cuda')
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
 
-# @spaces.GPU
-# def generate(prompt):
-#     return pipe(prompt).images
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
 
-# gr.Interface(
-#     fn=generate,
-#     inputs=gr.Text(),
-#     outputs=gr.Gallery(),
-# ).launch()
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()