Update tool/orbital.py

tool/orbital.py

@@ -63,9 +63,9 @@ def main(sm):
         n_features=10
         
         model = RandomForestRegressor(n_estimators=100)   
-        load_homo = pickle.load(open(r"tool/orbital/homo.pkl", 'rb'))
-        load_lumo = pickle.load(open(r"tool/orbital/lumo.pkl", 'rb'))
-     #   model = load_model('C:/Users/sunjinyu/Desktop/FingerID Reference/drug-likeness/CNN/single_model.h5')
+        load_homo = pickle.load(open(r"tool/orbital/homo.dat", 'rb'))
+        load_lumo = pickle.load(open(r"tool/orbital/lumo.dat", 'rb'))
+ 
         Y_homo= load_homo.predict(X_test)
         Y_lumo = load_lumo.predict(X_test)
         homo =  float(Y_homo)