demo 1

app.py

@@ -4,41 +4,30 @@ import platform
 import os;
 import socket;
 import torch
-import torch.nn as nn
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-class SimpleModel(nn.Module):
-    def __init__(self):
-        super(SimpleModel, self).__init__()
 
-    def forward(self, x):
-        pass
-        
-    @property
-    def device(self):
-       return next(model.parameters()).device
 
-model = SimpleModel().to(device);
-
-@torch.inference_mode()
-def sysinfo():
+def sysinfo(device):
+    RandomTensor = torch.randn(1, 2)  # Example audio tensor
+    tensorExample = RandomTensor.to(device)
     
     return f"""
        hostname: {platform.node()} {socket.gethostname()}
        device: {device}
-       model device: {model.device}
        tensor: {tensorExample}
     """;
 
 @spaces.GPU
 def gpu():
-     return sysinfo();
 
+     return sysinfo("cuda:0");
 
-def nogpu():
 
-    return sysinfo();
+def nogpu():
+    
+    return sysinfo("cpu");