Update app.py

app.py

@@ -10,7 +10,8 @@ class_names = ["pizza", "steak", "sushi"]
 
 ### 2 model and transform preparation###
 effnetb2_loaded, transforms = create_effnetb2_model(num_classes=len(class_names))
-effnetb2_loaded.load_state_dict("11-model_deployment_effnetb2.pth",map_location="cpu")
+effnetb2_loaded.load_state_dict("11-model_deployment_effnetb2.pth")
+effnetb2_loaded.to("cpu")
 
 
 ### 3 we need a predict function###