sharktide commited on
Commit
dab6870
·
verified ·
1 Parent(s): d9835d5

Update load.py

Browse files
Files changed (1) hide show
  1. load.py +9 -2
load.py CHANGED
@@ -5,7 +5,8 @@ hft.download_model("sharktide", "FireNet")
5
  hft.download_model("sharktide", "FireTrustNet")
6
  hft.download_model("sharktide", "FV-FloodNet")
7
  hft.download_model("sharktide", "FV-FloodTrustNet")
8
-
 
9
 
10
  FireNet = hft.load_model("sharktide", "FireNet", "tf_model.h5", True)
11
 
@@ -17,4 +18,10 @@ FloodNet = hft.load_model("sharktide", "FV-FloodNet", "tf_model.h5", True)
17
 
18
  FloodTrustNet = hft.load_model("sharktide", "FV-FloodTrustNet", "tf_model.h5", True)
19
 
20
- FloodScaler = joblib.load("scalers/FV-floodtrust_scaler.pkl")
 
 
 
 
 
 
 
5
  hft.download_model("sharktide", "FireTrustNet")
6
  hft.download_model("sharktide", "FV-FloodNet")
7
  hft.download_model("sharktide", "FV-FloodTrustNet")
8
+ hft.download_model("sharktide", "PV-FloodNet")
9
+ hft.download_model("sharktide", "PV-FloodTrustNet")
10
 
11
  FireNet = hft.load_model("sharktide", "FireNet", "tf_model.h5", True)
12
 
 
18
 
19
  FloodTrustNet = hft.load_model("sharktide", "FV-FloodTrustNet", "tf_model.h5", True)
20
 
21
+ FloodScaler = joblib.load("scalers/FV-floodtrust_scaler.pkl")
22
+
23
+ PV_FloodNet = hft.load_model("sharktide", "PV-FloodNet", "tf_model.h5", True)
24
+
25
+ PV_FloodTrustNet = hft.load_model("sharktide", "PV-FloodTrustNet", "tf_model.h5", True)
26
+
27
+ PV_FloodScaler = joblib.load("scalers/PV-floodtrust_scaler.pkl")