Spaces:

abdelac
/

FakeImageModelKeras

Sleeping

eesfeg commited on 25 days ago

Commit

0aa3dbd

1 Parent(s): 4351477

docker

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ model_path = hf_hub_download(
     filename="feature_extractor.h5"
 )
-feature_extractor = load_model(model_path)
 #feature_extractor = tf.keras.models.load_model(model_path)
 print("Model loaded successfully!")

     filename="feature_extractor.h5"
 )
+feature_extractor = load_model(model_path, compile=False, safe_mode=False)
 #feature_extractor = tf.keras.models.load_model(model_path)
 print("Model loaded successfully!")