update gdrive url

model.py

@@ -137,6 +137,7 @@ def check_file(image_path):
                 outputs = model(images)
                 _, predicted = torch.max(outputs.data, 1)
                 output = predicted
+                return predicted
 
     def remove_module_from_checkpoint(checkpoint):
         new_state_dict = {}
@@ -154,5 +155,5 @@ def check_file(image_path):
     model.eval()
     model.to(device)
     model = nn.DataParallel(model)
-    test_model(model, test_loader, device)
+    output = test_model(model, test_loader, device)
     return output