yolac commited on
Commit
6153dd4
·
verified ·
1 Parent(s): f4aea2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from PIL import Image
10
  app = FastAPI()
11
 
12
  # Path to the trained model checkpoint
13
- MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/BacterialMorphologyClassification_model.ipynb"
14
 
15
  # Load the pretrained MobileNetV2 model and modify the classifier
16
  model = models.mobilenet_v2(weights=None) # No initial weights
 
10
  app = FastAPI()
11
 
12
  # Path to the trained model checkpoint
13
+ MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/blob/main/BacterialMorphologyClassification_model.ipynb"
14
 
15
  # Load the pretrained MobileNetV2 model and modify the classifier
16
  model = models.mobilenet_v2(weights=None) # No initial weights