rrg92 commited on
Commit
9b6e7fb
·
1 Parent(s): 3dbc4a6
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,8 +12,10 @@ if not torch.cuda.is_available() and device == "cuda":
12
 
13
  print("DEVICE: ", device);
14
 
 
 
 
15
  def sysinfo(device):
16
- RandomTensor = torch.randn(1, 2) # Example audio tensor
17
  tensorExample = RandomTensor.to(device)
18
 
19
  tocpu = tensorExample.cpu().squeeze().half().tolist()
 
12
 
13
  print("DEVICE: ", device);
14
 
15
+ RandomTensor = torch.randn(1, 2) # Example audio tensor
16
+ RandomTensor.to(device)
17
+
18
  def sysinfo(device):
 
19
  tensorExample = RandomTensor.to(device)
20
 
21
  tocpu = tensorExample.cpu().squeeze().half().tolist()