Update app.py

app.py

@@ -25,7 +25,7 @@ class Classifier:
         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"
+            self.device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
         else:
             self.device = device