Spaces:

cbensimon
/

zero-gpu-test0

Sleeping

cbensimon HF Staff commited on Jul 6, 2023

Commit

f948f08

1 Parent(s): 43b8d8c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,10 @@
 import gradio as gr
 def greet(name):
-    return "Hello " + name + "!!"
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()

 import gradio as gr
+import spaces
+import torch
+@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()