Update app.py

app.py

@@ -1,16 +1,21 @@
 import gradio as gr
 import tensorflow as tf
-from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing import image
 import numpy as np
-from huggingface_hub import from_pretrained_keras
+from huggingface_hub import hf_hub_download
+import os
+
+# Download the models
+model1_path = hf_hub_download(repo_id="arsath-sm/face_classification_model1", filename="model.h5")
+model2_path = hf_hub_download(repo_id="arsath-sm/face_classification_model2", filename="model.h5")
 
 # Load the models
-model1 = tf.keras.layers.TFSMLayer("arsath-sm/face_classification_model1", call_endpoint='serving_default')
-model2 = tf.keras.layers.TFSMLayer("arsath-sm/face_classification_model2", call_endpoint='serving_default')
+model1 = tf.keras.models.load_model(model1_path)
+model2 = tf.keras.models.load_model(model2_path)
 
 # Preprocess the image
 def preprocess_image(img):
+    img = img.resize((224, 224))  # Adjust size as needed
     img = image.img_to_array(img)
     img = np.expand_dims(img, axis=0)
     img = img / 255.0