thejagstudio commited on
Commit
44490bb
·
verified ·
1 Parent(s): 5ea8efc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,10 +12,10 @@ import json
12
  app = Flask(__name__)
13
 
14
  # Load model and data at startup
15
- # model = tf.keras.models.load_model("/data/app/sneaker_category_predictor_v2.h5")
16
- model_path = hf_hub_download(repo_id="thejagstudio/SneakerAI", filename="sneaker_category_predictor_v2.h5", repo_type="model")
17
  # model = tf.keras.models.load_model(model_path)
18
- model = load_model(model_path)
19
 
20
  # Define expected columns for one-hot encoding
21
  with open("metadata.json", "r") as f:
 
12
  app = Flask(__name__)
13
 
14
  # Load model and data at startup
15
+ model = tf.keras.models.load_model("./sneaker_category_predictor_v2.keras")
16
+ # model_path = hf_hub_download(repo_id="thejagstudio/SneakerAI", filename="sneaker_category_predictor_v2.h5", repo_type="model")
17
  # model = tf.keras.models.load_model(model_path)
18
+ # model = load_model(model_path)
19
 
20
  # Define expected columns for one-hot encoding
21
  with open("metadata.json", "r") as f: