Hugging Face's logo

Spaces:
yolac
/
BacterialMorphologyClassification
Running

App Files Community
yolac commited on
Commit
919808e
·
verified ·
1 Parent(s): 007a13c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ class BacterialMorphologyClassifier(nn.Module):
35
  return x
36
 
37
  # Load the model and weights
38
- MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/model.pth"
39
  logging.debug("Starting model loading...")
40
  try:
41
  model = BacterialMorphologyClassifier()
 
35
  return x
36
 
37
  # Load the model and weights
38
+ MODEL_PATH = "https://huggingface.co/yolac/BacterialMorphologyClassification/resolve/main/model.keras"
39
  logging.debug("Starting model loading...")
40
  try:
41
  model = BacterialMorphologyClassifier()