AliAmr0 commited on
Commit
ae3babb
·
verified ·
1 Parent(s): b8d882d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  from huggingface_hub import hf_hub_download
8
 
9
  # Download the model from Hugging Face
10
- model_path = hf_hub_download(repo_id="AliAmr0/Kidney-Classification-Using-Resnet50", filename="model.h5")
11
  model = tf.keras.models.load_model(model_path)
12
 
13
  # Class labels (change based on your model's labels)
 
7
  from huggingface_hub import hf_hub_download
8
 
9
  # Download the model from Hugging Face
10
+ model_path = hf_hub_download(repo_id="AliAmr0/Kidney-Classification-Using-Resnet50", filename="resnet50_kidney_ct_augmented.h5")
11
  model = tf.keras.models.load_model(model_path)
12
 
13
  # Class labels (change based on your model's labels)