update

app.py

@@ -45,7 +45,8 @@ model = ResNet_CSRA(num_heads=1, lam=0.1, num_classes=20)
 normalize = transforms.Normalize(mean=[0, 0, 0], std=[1, 1, 1])
 model.to(DEVICE)
 print("Loading weights from {}".format("./models/msl_c_voc.pth"))
-model.load_state_dict(torch.load("./models/msl_c_voc.pth"), map_location=torch.device('cpu'))
+model.load_state_dict(torch.load("./models/msl_c_voc.pth", map_location=torch.device("cpu")))
+model.eval()
 
 # Inference!
 def inference(img_path):
@@ -64,7 +65,6 @@ def inference(img_path):
 
     # Predict
     result = []
-    model.eval()
     with torch.no_grad():
         image = image.to(DEVICE)
         logit = model(image).squeeze(0)