update gdrive url

model.py

@@ -147,7 +147,7 @@ def check_file(image_path):
 
     model = HybridCNNViT(3, 2)
     checkpoint = torch.load(
-        "/home/user/app/checkpoint32.pth", weights_only=False)
+        "/home/user/app/checkpoint32.pth", weights_only=False, map_location=torch.device('cpu'))
     checkpoint = remove_module_from_checkpoint(checkpoint)
     model.load_state_dict(checkpoint['model_state_dict'])
     # optimizer.load_state_dict(checkpoint['optimizer_state_dict'])