Update app.py

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import hf_hub_download
 import cv2
 import numpy as np
 import tensorflow as tf
+from keras.models import load_model
 import pandas as pd
 import base64
 import json
@@ -13,7 +14,8 @@ app = Flask(__name__)
 # Load model and data at startup
 # model = tf.keras.models.load_model("/data/app/sneaker_category_predictor_v2.h5")
 model_path = hf_hub_download(repo_id="thejagstudio/SneakerAI", filename="sneaker_category_predictor_v2.h5", repo_type="model")
-model = tf.keras.models.load_model(model_path, compile=False)
+# model = tf.keras.models.load_model(model_path)
+model = load_model(model_path)
 
 # Define expected columns for one-hot encoding
 with open("metadata.json", "r") as f: