Hugging Face's logo

Spaces:
SaniaE
/
Car_Damage_Detection_API
Sleeping

App Files Community
SaniaE commited on
Commit
a60a006
·
verified ·
1 Parent(s): a2b5329

updated model loading

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ CNN_MODEL_PATH = os.path.join(model_dir, "cnn_filter.h5")
33
 
34
  # 3. Load Models
35
  # Load CNN Filter
36
- cnn_filter = tf.keras.models.load_model(CNN_MODEL_PATH)
37
 
38
  # Load Object Detection Model
39
  configs = config_util.get_configs_from_pipeline_file(PIPELINE_CONFIG)
 
33
 
34
  # 3. Load Models
35
  # Load CNN Filter
36
+ cnn_filter = tf.keras.models.load_model(CNN_MODEL_PATH, compile=False)
37
 
38
  # Load Object Detection Model
39
  configs = config_util.get_configs_from_pipeline_file(PIPELINE_CONFIG)