Update app.py

app.py

@@ -27,10 +27,12 @@ def get_model(num_classes):
     return model
 
 model = get_model(num_classes=len(class_names))
-model.load_state_dict(torch.load(MODEL_PATH, map_location=DEVICE))
+state_dict = torch.load(MODEL_PATH, map_location=DEVICE, weights_only=False)
+model.load_state_dict(state_dict)
 model.eval()
 model.to(DEVICE)
 
+
 def predict_car(img):
     image = img.convert("RGB")
     input_tensor = val_transform(image).unsqueeze(0).to(DEVICE)