Update app

app.py

@@ -40,6 +40,7 @@ warnings.filterwarnings("ignore", category=UserWarning)
 CHECKPOINT_PATH = "model_rgb.pth"  
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 use_cuda = torch.cuda.is_available()
+print("Use cuda status: ", use_cuda)
 batch_size = 12
 fps = 25
 n_negative_samples = 100
@@ -537,11 +538,11 @@ def load_checkpoint(path, model):
 	'''	
 
 	# Load the checkpoint
-	# checkpoint = torch.load(path, map_location="cpu")
-	if use_cuda:
-		checkpoint = torch.load(path)
-	else:
-		checkpoint = torch.load(path, map_location="cpu")
+	checkpoint = torch.load(path, map_location="cpu")
+	# if use_cuda:
+	# 	checkpoint = torch.load(path)
+	# else:
+	# 	checkpoint = torch.load(path, map_location="cpu")
 	
 	s = checkpoint["state_dict"]
 	new_s = {}