Update app.py

app.py

@@ -1,12 +1,13 @@
 import streamlit as st
 import tensorflow as tf
-from tensorflow.keras.applications.resnet50 import ResNet50, preprocess_input
+from tensorflow.keras.applications.resnet50 import preprocess_input
 from tensorflow.keras.preprocessing import image
 import numpy as np
 from PIL import Image
 
-# Load pre-trained ResNet50 model
-model = tf.keras.models.load_model("resnet50_kidney_ct_augmented.h5")  # Update this path if you are using a .pb file
+# Download the model from Hugging Face
+model_path = hf_hub_download(repo_id="AliAmr0/Kidney-Classification-Using-Resnet50", filename="model.h5")
+model = tf.keras.models.load_model(model_path)
 
 # Class labels (change based on your model's labels)
 labels = ["Cyst", "Normal", "Stone", "Tumor"]