Spaces:

raphael-gl
/

test-zero

Sleeping

raphael-gl HF Staff commited on Oct 23

Commit

4380bef

verified ·

1 Parent(s): 45b1ca5

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(duration=1600)
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"

 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
+@spaces.GPU(duration=220)
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"