rrg92 commited on
Commit
bf3cf2c
·
1 Parent(s): aa239ce
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -15,6 +15,7 @@ class SimpleModel(nn.Module):
15
  def forward(self, x):
16
  pass
17
 
 
18
  def device():
19
  return next(model.parameters()).device
20
 
@@ -22,8 +23,6 @@ model = SimpleModel().to(device);
22
 
23
  @torch.inference_mode()
24
  def sysinfo():
25
- RandomTensor = torch.randn(1, 2) # Example audio tensor
26
- tensorExample = RandomTensor.to(device)
27
 
28
  return f"""
29
  hostname: {platform.node()} {socket.gethostname()}
 
15
  def forward(self, x):
16
  pass
17
 
18
+ @property
19
  def device():
20
  return next(model.parameters()).device
21
 
 
23
 
24
  @torch.inference_mode()
25
  def sysinfo():
 
 
26
 
27
  return f"""
28
  hostname: {platform.node()} {socket.gethostname()}