Update app.py

app.py

@@ -1,11 +1,18 @@
 import gradio as gr
 import tensorflow as tf
-from tensorflow.keras.preprocessing import image
 import numpy as np
-from huggingface_hub import from_pretrained_keras
+from tensorflow.keras.preprocessing import image
+from huggingface_hub import hf_hub_download
+
+# Define model repository and filename
+MODEL_REPO = "mramjad/cats_dogs_model"
+MODEL_FILENAME = "cats_dogs_model.h5"
+
+# Download the model file
+model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
 
-# Load the trained model from Hugging Face Hub
-model = from_pretrained_keras("mramjad/cats_dogs_model")
+# Load the model (ensure it's in `.h5` format)
+model = tf.keras.models.load_model(model_path)
 model.compile(optimizer="adam", loss="categorical_crossentropy")
 
 # Class labels
@@ -34,5 +41,5 @@ iface = gr.Interface(
     allow_flagging="never"
 )
 
-# 🚀 **Launch Fix (No `share=True`)**
-iface.launch(server_name="0.0.0.0", server_port=7860)
+# Launch the Gradio app
+iface.launch(server_name="0.0.0.0", server_port=7860)