Spaces:

woshixuhao
/

Rf_prediction

Running

woshixuhao commited on May 27, 2023

Commit

f99f51e

1 Parent(s): b5b9551

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -178,10 +178,10 @@ class Model_ML():
             Net = ANN(self.X_test.shape[1], self.NN_hidden_neuron, 1, config=self.config).to(self.device)
             #model_LGB = joblib.load('model_LGB.pkl')
-            wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_LGB.pkl')
-            wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_XGB.pkl')
-            wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_RF.pkl')
-            wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_ANN.pkl')
             model_LGB = joblib.load('model_LGB.pkl')
             model_XGB = joblib.load('model_XGB.pkl')
             model_RF = joblib.load('model_RF.pkl')

             Net = ANN(self.X_test.shape[1], self.NN_hidden_neuron, 1, config=self.config).to(self.device)
             #model_LGB = joblib.load('model_LGB.pkl')
+            #wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_LGB.pkl')
+            #wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_XGB.pkl')
+            #wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_RF.pkl')
+            #wget.download('https://huggingface.co/woshixuhao/Rf_prediction/resolve/main/model_ANN.pkl')
             model_LGB = joblib.load('model_LGB.pkl')
             model_XGB = joblib.load('model_XGB.pkl')
             model_RF = joblib.load('model_RF.pkl')