change model weight save path

app.py

@@ -13,6 +13,7 @@ import urllib.request
 
 MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
 MODEL_PATH = "/tmp/best_model.pth" 
+urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
 
 test_transforms = A.Compose(
     [
@@ -41,8 +42,13 @@ model.fc = nn.Sequential(
 )
 
 device = torch.device('cpu')
-urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
-model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
+
+try:
+    model.load_state_dict(torch.load(MODEL_PATH, map_location=device))
+    print('model load complete')
+except:
+    print('CANNOT load model weight')
+    
 model.eval()
 for _, p in model.named_parameters():
     p.requires_grad = False