Update app.py

app.py

@@ -1,5 +1,5 @@
 from flask import Flask, request, render_template, jsonify
-# from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download
 import cv2
 import numpy as np
 import tensorflow as tf
@@ -11,9 +11,9 @@ import json
 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")
-# model = tf.keras.models.load_model(model_path)
+# 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)
 
 # Define expected columns for one-hot encoding
 with open("metadata.json", "r") as f: