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 tensorflow import keras
 import pandas as pd
 import base64
 import json
@@ -11,7 +12,8 @@ import json
 app = Flask(__name__)
 
 # Load model and data at startup
-model = tf.keras.models.load_model("./sneaker_category_predictor_v2.keras")
+model = keras.models.load_model("./sneaker_category_predictor_v2.keras")
+# model = tf.keras.models.load_model("./sneaker_category_predictor_v2.keras")
 # 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)
 # model = load_model(model_path)