aaravlovescodes commited on
Commit
2801237
·
verified ·
1 Parent(s): 281deee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def load_model():
55
  from huggingface_hub import from_pretrained_keras, hf_hub_download
56
 
57
  # Download the model files directly
58
- model_path = hf_hub_download(repo_id="ryefoxlime/PneumoniaDetection", filename="model.keras")
59
  model = keras.models.load_model(model_path)
60
  return model
61
  except Exception as e:
 
55
  from huggingface_hub import from_pretrained_keras, hf_hub_download
56
 
57
  # Download the model files directly
58
+ model_path = hf_hub_download(repo_id="ryefoxlime/PneumoniaDetection", filename="saved_model.pb")
59
  model = keras.models.load_model(model_path)
60
  return model
61
  except Exception as e: