Update app.py

app.py

@@ -24,7 +24,7 @@ custom_model_file = io.BytesIO(custom_model_content)
 custom_model_state_dict = torch.load(custom_model_file, map_location=torch.device('cpu'))
 
 # Create a new instance of your custom model
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='/content/efficientnet_b3.pt')
+model = torch.hub.load('ultralytics/yolov5', 'custom', path_or_model=efficientnet_model)
 
 # Load your custom model into the new instance
 model.load_state_dict(custom_model_state_dict)