demo 1

app.py

@@ -10,12 +10,12 @@ device = "cuda";
 if not torch.cuda.is_available() and device == "cuda":
     raise RuntimeError("CUDA device unavailable, please use Dockerfile.cpu instead.") 
 
-print("DEVICE: ", device);
-
 RandomTensor = torch.randn(1, 2)  # Example audio tensor
+
+print("Putting Tensor in device ...", device);
 RandomTensor.to(device)
 
-def sysinfo(device):
+def sysinfo():
     tensorExample = RandomTensor.to(device)
     
     tocpu = tensorExample.cpu().squeeze().half().tolist()
@@ -30,12 +30,12 @@ def sysinfo(device):
 @spaces.GPU
 def gpu():
 
-     return sysinfo("cuda:0");
+     return sysinfo();
 
 
 def nogpu():
     
-    return sysinfo("cpu");
+    return sysinfo();