docker

app.py

@@ -14,11 +14,13 @@ import cv2
 from huggingface_hub import hf_hub_download
 import tensorflow as tf
 
-feature_extractor = hf_hub_download(
+
+model_path = hf_hub_download(
     repo_id="abdelac/FakeImageModelKeras",
     filename="feature_extractor.keras"
 )
 
+feature_extractor = load_model(model_path)
 #feature_extractor = tf.keras.models.load_model(model_path)
 print("Model loaded successfully!")