Update app.py
Browse files
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)
|