Update app.py

app.py

@@ -10,7 +10,7 @@ from io import BytesIO
 app = FastAPI()
 
 # Load the model
-MODEL_PATH = "BacterialMorphologyClassification_model.ipynb"
+MODEL_PATH = "BacterialMorphologyClassification_model.pth.ipynb"
 model = models.mobilenet_v2(weights=None)
 model.classifier[1] = nn.Linear(model.last_channel, 3)
 model.load_state_dict(torch.hub.load_state_dict_from_url(MODEL_PATH, map_location=torch.device('cpu')))