gmedin commited on
Commit
98f49ee
·
verified ·
1 Parent(s): 8d70952

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def load_graph_from_hf(brand):
31
  # Download the file from HuggingFace Hub
32
  HF_REPO = f'{brand}-graph'
33
  cache_dir = '/tmp'
34
- file_path = hf_hub_download(repo_id=HF_REPO, filename=BRAND_GRAPHS[brand], token=AUTH_TOKEN, cache_dir=cache_dir)
35
  # Load the graph
36
  with open(file_path, 'rb') as f:
37
  return pickle.load(f)
 
31
  # Download the file from HuggingFace Hub
32
  HF_REPO = f'{brand}-graph'
33
  cache_dir = '/tmp'
34
+ file_path = hf_hub_download(repo_id=HF_REPO, filename=BRAND_GRAPHS[brand], token=AUTH_TOKEN, cache_dir=cache_dir, repo_type='model')
35
  # Load the graph
36
  with open(file_path, 'rb') as f:
37
  return pickle.load(f)