Spaces:

yolac
/

BacterialMorphologyClassification

Running

yolac commited on Dec 9, 2024

Commit

3d4277c

verified ·

1 Parent(s): f1573d9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from io import BytesIO
 app = FastAPI()
 # Load the model
-MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/blob/main/model.pth"
 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')))

 app = FastAPI()
 # Load the model
+MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/model.pth"
 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')))