IFMedTechdemo commited on
Commit
bd8f002
·
verified ·
1 Parent(s): a8de3ef

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ def load_model(token):
23
  """Download and load ONNX model"""
24
  model_path = hf_hub_download(
25
  repo_id=REPO_ID,
26
- filename="model/wrinkle_model.onnx", # Adjust path if needed
27
  token=token
28
  )
29
 
 
23
  """Download and load ONNX model"""
24
  model_path = hf_hub_download(
25
  repo_id=REPO_ID,
26
+ filename="model/wrinkle_model.onnx.data", # Adjust path if needed
27
  token=token
28
  )
29