demo 1

app.py

@@ -16,7 +16,7 @@ class SimpleModel(nn.Module):
         pass
         
     @property
-    def device():
+    def device(self):
        return next(model.parameters()).device
 
 model = SimpleModel().to(device);