Update app.py

app.py

@@ -22,8 +22,13 @@ class Classifier:
 
     @spaces.GPU(duration=60)
     def __init__(
-        self, model_path="Litton-7type-visual-landscape-model.pth", device="cuda:0"
+        self, model_path="Litton-7type-visual-landscape-model.pth", device=None
     ):
+        if device is None:
+            self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
+        else:
+            self.device = device
+
         self.device = device
         self.model = torch.load(
             model_path, map_location=self.device, weights_only=False