weidai00 commited on
Commit
6ae64b3
·
verified ·
1 Parent(s): 6df031f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -300,7 +300,8 @@ def segment_by_out_test(image,model_name):
300
  model_path = hf_hub_download(
301
  repo_id="weidai00/RIP-AV-sulab", # 模型库的名字
302
  filename=f"G_{model_name}.pkl", # 文件名
303
- repo_type="model" # 模型库必须写 repo_type
 
304
  )
305
  cfg.set_dataset(model_name)
306
  if image is None:
 
300
  model_path = hf_hub_download(
301
  repo_id="weidai00/RIP-AV-sulab", # 模型库的名字
302
  filename=f"G_{model_name}.pkl", # 文件名
303
+ repo_type="model", # 模型库必须写 repo_type
304
+ use_auth_token=True
305
  )
306
  cfg.set_dataset(model_name)
307
  if image is None: