suko commited on
Commit
3fec243
1 Parent(s): a06b7be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ class ONNXModel:
8
  def __init__(self, dir_path) -> None:
9
  """Method to get name of model file. Assumes model is in the parent directory for script."""
10
  model_dir = os.path.dirname(dir_path)
11
- with open(os.path.join(model_dir, "signature.json"), "r") as f:
12
  self.signature = json.load(f)
13
  self.model_file = os.path.join(model_dir, self.signature.get("filename"))
14
  if not os.path.isfile(self.model_file):
 
8
  def __init__(self, dir_path) -> None:
9
  """Method to get name of model file. Assumes model is in the parent directory for script."""
10
  model_dir = os.path.dirname(dir_path)
11
+ with open(os.path.join(model_dir, "/signature.json"), "r") as f:
12
  self.signature = json.load(f)
13
  self.model_file = os.path.join(model_dir, self.signature.get("filename"))
14
  if not os.path.isfile(self.model_file):