3v324v23 commited on
Commit
c3c28df
·
1 Parent(s): 6d359cf
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,7 +77,7 @@ def main():
77
  st.markdown("### Setup")
78
  st.markdown("\n Please fill the following fields with the path to the NSRwH and NSR models. Instruction on how to get or \
79
  train these models can be found in the README.md file")
80
- nsrwh = st.text_input("Path to the NSRwH model", "ControllableNeuralSymbolicRegressionWeights/nsrwh_200000000_epoch=149.ckpt")
81
 
82
  metadata = load_metadata_hdf5(Path("configs"))
83
  # Retrofit word2id if there is conditioning
@@ -102,7 +102,7 @@ def main():
102
  negative_pool = prepare_negative_pool(cfg)
103
 
104
  st.markdown("### Define the equation to test")
105
- equation_examples = ["0.01*x_1**2+x_2+exp(x_3+x_4)", "sin(x_1)+sqrt(x_2)+sin(x_3/2+x_4)", "x_1**(1/2)+0.3*x_2**2+x_3**2", "exp(0.043*sin(x_1*x_2))+x_3+3*x_4", "0.2*x_1**2+x_2**2+(x_3*x_4**2)", "other"]
106
  eq_string = st.selectbox("Select an equation or select on 'other' to write your own equation to test", equation_examples, index=4)
107
 
108
  if eq_string == "other":
 
77
  st.markdown("### Setup")
78
  st.markdown("\n Please fill the following fields with the path to the NSRwH and NSR models. Instruction on how to get or \
79
  train these models can be found in the README.md file")
80
+ nsrwh = "ControllableNeuralSymbolicRegressionWeights/nsrwh_200000000_epoch=149.ckpt" #st.text_input("Path to the NSRwH model", "ControllableNeuralSymbolicRegressionWeights/nsrwh_200000000_epoch=149.ckpt")
81
 
82
  metadata = load_metadata_hdf5(Path("configs"))
83
  # Retrofit word2id if there is conditioning
 
102
  negative_pool = prepare_negative_pool(cfg)
103
 
104
  st.markdown("### Define the equation to test")
105
+ equation_examples = ["0.01*x_1**2+x_2+exp(x_3)+x_4", "sin(x_1)+sqrt(x_2)+sin(x_3/2+x_4)", "x_1**(1/2)+0.3*x_2**2+x_3**2", "exp(0.043*sin(x_1*x_2))+x_3+3*x_4", "0.2*x_1**2+x_2**2+(x_3*x_4**2)", "other"]
106
  eq_string = st.selectbox("Select an equation or select on 'other' to write your own equation to test", equation_examples, index=4)
107
 
108
  if eq_string == "other":