DoctorSlimm commited on
Commit
abe7e15
·
verified ·
1 Parent(s): fbbcfb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)