kaveh commited on
Commit
8a0f092
·
1 Parent(s): e6b36e6

updated names based on Mari suggestion

Browse files
Files changed (1) hide show
  1. predictor.py +1 -1
predictor.py CHANGED
@@ -129,7 +129,7 @@ class S2FPredictor:
129
  self._use_tanh_output = model_type == "single_cell" # single_cell uses tanh, spheroid uses sigmoid
130
  self.config_path = os.path.join(S2F_ROOT, "config", "substrate_settings.json")
131
 
132
- def predict(self, image_path=None, image_array=None, substrate="fibroblasts_PDMS",
133
  substrate_config=None):
134
  """
135
  Run prediction on an image.
 
129
  self._use_tanh_output = model_type == "single_cell" # single_cell uses tanh, spheroid uses sigmoid
130
  self.config_path = os.path.join(S2F_ROOT, "config", "substrate_settings.json")
131
 
132
+ def predict(self, image_path=None, image_array=None, substrate="Fibroblasts_Fibronectin_6KPa",
133
  substrate_config=None):
134
  """
135
  Run prediction on an image.