Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,8 +20,8 @@ def greet(prompts, separator):
|
|
| 20 |
from huggingface_hub import snapshot_download, login
|
| 21 |
|
| 22 |
LLM_MODEL_ID = "mozart-ai/llm-schema-extractor"
|
| 23 |
-
|
| 24 |
-
fp = snapshot_download(LLM_MODEL_ID, token=os.getenv('HF_TOKEN'))
|
| 25 |
move_cache()
|
| 26 |
|
| 27 |
model = LLM(fp)
|
|
|
|
| 20 |
from huggingface_hub import snapshot_download, login
|
| 21 |
|
| 22 |
LLM_MODEL_ID = "mozart-ai/llm-schema-extractor"
|
| 23 |
+
os.environ['HF_HUB_ENABLE_HF_TRANSFER'] = '1'
|
| 24 |
+
fp = snapshot_download(LLM_MODEL_ID, token=os.getenv('HF_TOKEN'), revision='main')
|
| 25 |
move_cache()
|
| 26 |
|
| 27 |
model = LLM(fp)
|