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()