Hugging Face's logo

Spaces:
sharktide
/
strike
Running

App Files Community
sharktide commited on
Commit
8b3a2bc
·
verified ·
1 Parent(s): a199ad7

Update load.py

Browse files
Files changed (1) hide show
  1. load.py +1 -1
load.py CHANGED
@@ -58,7 +58,7 @@ FloodNet = hft.load_model("sharktide", "FV-FloodNet", "tf_model.h5", True)
58
  FloodTrustNet = hft.load_model("sharktide", "FV-FloodTrustNet", "tf_model.h5", True)
59
  FloodScaler = joblib.load("scalers/FV-floodtrust_scaler.pkl")
60
 
61
- get_path = lambda usr, model: (str(hft.get_model_folder(usr, model)) + "tf_model.h5")
62
  PV_FloodNet = tf.keras.models.load_model(get_path("sharktide", "PV-FloodNet"), safe_mode=False, custom_objects=CUSTOM_OBJECTS)
63
  PV_FloodTrustNet = hft.load_model("sharktide", "PV-FloodTrustNet", "tf_model.h5", True)
64
  PV_FloodScaler = joblib.load("scalers/PV-floodtrust_scaler.pkl")
 
58
  FloodTrustNet = hft.load_model("sharktide", "FV-FloodTrustNet", "tf_model.h5", True)
59
  FloodScaler = joblib.load("scalers/FV-floodtrust_scaler.pkl")
60
 
61
+ get_path = lambda usr, model: (str(hft.get_model_folder(usr, model)) + "/tf_model.h5")
62
  PV_FloodNet = tf.keras.models.load_model(get_path("sharktide", "PV-FloodNet"), safe_mode=False, custom_objects=CUSTOM_OBJECTS)
63
  PV_FloodTrustNet = hft.load_model("sharktide", "PV-FloodTrustNet", "tf_model.h5", True)
64
  PV_FloodScaler = joblib.load("scalers/PV-floodtrust_scaler.pkl")