Update app.py

app.py

@@ -5,10 +5,11 @@ import numpy as np
 from huggingface_hub import hf_hub_download
 import os
 
-
+def load_model_from_hub(repo_id, filename):
+    model_path = hf_hub_download(repo_id=repo_id, filename=filename)
+    return tf.keras.models.load_model(model_path)
 
 # Load the models
-
 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")