Spaces:

cbensimon
/

gradio-state

Sleeping

cbensimon HF Staff commited on Sep 20, 2023

Commit

12a3156

1 Parent(s): 00bf99a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
-@spaces.GPU
 def greet(n, name):
     print(name)
     n += 1

 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
+@spaces.GPU(duration=1)
 def greet(n, name):
     print(name)
     n += 1