Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def load_graph_from_hf(brand):
|
|
| 29 |
"""
|
| 30 |
try:
|
| 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
|
|
|
|
| 29 |
"""
|
| 30 |
try:
|
| 31 |
# Download the file from HuggingFace Hub
|
| 32 |
+
HF_REPO = f'MusoraProductDepartment/{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
|