update return msg

model.py

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