Update app.py

app.py

@@ -5,13 +5,12 @@ import numpy as np
 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.models.load_model(model1_path)
-model2 = tf.keras.models.load_model(model2_path)
+
+model1 = load_model_from_hub("arsath-sm/face_classification_model1", "face_classification_model1.h5")
+model2 = load_model_from_hub("arsath-sm/face_classification_model2", "face_classification_model2.h5")
 
 # Preprocess the image
 def preprocess_image(img):