thejagstudio commited on
Commit
c1ecdf4
·
verified ·
1 Parent(s): 6ff8643

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import json
11
  app = Flask(__name__)
12
 
13
  # Load model and data at startup
14
- model = tf.keras.models.load_model("sneaker_category_predictor_v2.h5")
15
  # model_path = hf_hub_download(repo_id="thejagstudio/SneakerAI", filename="sneaker_category_predictor_v2.h5")
16
  # model = tf.keras.models.load_model(model_path)
17
 
 
11
  app = Flask(__name__)
12
 
13
  # Load model and data at startup
14
+ model = tf.keras.models.load_model("/data/app/sneaker_category_predictor_v2.h5")
15
  # model_path = hf_hub_download(repo_id="thejagstudio/SneakerAI", filename="sneaker_category_predictor_v2.h5")
16
  # model = tf.keras.models.load_model(model_path)
17