Update app.py

app.py

@@ -41,8 +41,6 @@ css = """
 }
 """
 
-DEVICE = 'cuda' if torch.cuda.is_available() else 'mps'
-
 import json
 from EcoDepth.model import EcoDepth
 
@@ -59,14 +57,14 @@ args = Args()
 args_indoor = copy.deepcopy(args)
 args_indoor.no_of_classes = 100
 args_indoor.max_depth = 10
-model_indoor = EcoDepth(args_indoor).to(DEVICE).eval()
+model_indoor = EcoDepth(args_indoor).eval()
 model_str_indoor = f"{args.ckpt_path}/weights_indoor.ckpt"
 model_indoor.load_state_dict(torch.load(model_str_indoor, map_location="cpu", weights_only=True)["state_dict"])
 
 args_outdoor = copy.deepcopy(args)
 args_outdoor.no_of_classes = 200
 args_outdoor.max_depth = 80
-model_outdoor = EcoDepth(args_outdoor).to(DEVICE).eval()
+model_outdoor = EcoDepth(args_outdoor).eval()
 model_str_outdoor = f"{args.ckpt_path}/weights_outdoor.ckpt"
 model_outdoor.load_state_dict(torch.load(model_str_outdoor, map_location="cpu", weights_only=True)["state_dict"])