boryasbora commited on
Commit
d9e62b0
·
verified ·
1 Parent(s): c5a1b5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ from langchain_core.runnables import RunnableLambda
16
  from datetime import date
17
  from transformers import AutoModelForCausalLM, AutoTokenizer
18
  from setup import OLMO_MODEL
19
- model_path = OLMO_MODEL
 
20
 
21
  # Environment variables
22
  os.environ['LANGCHAIN_TRACING_V2'] = 'true'
 
16
  from datetime import date
17
  from transformers import AutoModelForCausalLM, AutoTokenizer
18
  from setup import OLMO_MODEL
19
+ from setup import download_olmo_model
20
+ model_path = download_olmo_model()
21
 
22
  # Environment variables
23
  os.environ['LANGCHAIN_TRACING_V2'] = 'true'