app.py

app.py

@@ -37,7 +37,7 @@ def score(input_img):
     class Config:
         im_size =  224
         batch_size = 1
-        num_workers = 8
+        num_workers = 0
         # }}}
     # 03. Define Dataset {{{
     class PetDataset(Dataset):
@@ -183,7 +183,6 @@ def score(input_img):
 
     test_preds = []
     test_preds_model = []
-    modelfiles = glob(Config.model_base_dir + Config_exp77.model_dir + Config.model_file_ext)
     test_preds_fold = []
     model = PetNet_exp77(model_name = 'beit_large_patch16_224', out_features = 1, inp_channels = 3, pretrained=False)
     model.load_state_dict(torch.load('beit_large_patch16_224_fold1_half.pth', map_location=torch.device('cpu')))