thejagstudio commited on
Commit
5092a44
·
verified ·
1 Parent(s): 01de232

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ app = Flask(__name__)
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", repo_type="model")
16
- model = tf.keras.models.load_model(model_path)
17
 
18
  # Define expected columns for one-hot encoding
19
  with open("metadata.json", "r") as f:
 
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", repo_type="model")
16
+ model = tf.keras.models.load_model(model_path, compile=False)
17
 
18
  # Define expected columns for one-hot encoding
19
  with open("metadata.json", "r") as f: