Update app.py

app.py

@@ -1,17 +1,16 @@
 import gradio as gr
+from huggingface_hub import from_pretrained_keras
 import tensorflow as tf
-from PIL import Image
 import numpy as np
 
-# Load the models
-model1 = tf.keras.models.load_model('face_detection_model1.h5')
-model2 = tf.keras.models.load_model('face_detection_model2.h5')
+# Load models from Hugging Face Hub
+model1 = from_pretrained_keras("arsath-sm/real-fake-face-detection-model1")
+model2 = from_pretrained_keras("arsath-sm/real-fake-face-detection-model2")
 
 def preprocess_image(image):
-    img = Image.fromarray(image.astype('uint8'), 'RGB')
-    img = img.resize((150, 150))
-    img_array = np.array(img) / 255.0
-    return np.expand_dims(img_array, axis=0)
+    img = tf.image.resize(image, (150, 150))
+    img = img / 255.0
+    return tf.expand_dims(img, 0)
 
 def predict_image(image):
     preprocessed_image = preprocess_image(image)