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 = '{brand}-graph'
|
| 33 |
file_path = hf_hub_download(repo_id=HF_REPO, filename=BRAND_GRAPHS[brand], use_auth_token=AUTH_TOKEN)
|
| 34 |
# Load the graph
|
| 35 |
with open(file_path, 'rb') as f:
|
|
|
|
| 29 |
"""
|
| 30 |
try:
|
| 31 |
# Download the file from HuggingFace Hub
|
| 32 |
+
HF_REPO = f'{brand}-graph'
|
| 33 |
file_path = hf_hub_download(repo_id=HF_REPO, filename=BRAND_GRAPHS[brand], use_auth_token=AUTH_TOKEN)
|
| 34 |
# Load the graph
|
| 35 |
with open(file_path, 'rb') as f:
|