Spaces:

boryasbora
/

chatbot_ohw_projects

Sleeping

boryasbora commited on Aug 29, 2024

Commit

a9e89f5

verified ·

1 Parent(s): bad6dd5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,6 +16,7 @@ from langchain_core.runnables import RunnableLambda
 from datetime import date
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from setup import OLMO_MODEL
 model_path = OLMO_MODEL
 # Environment variables

 from datetime import date
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from setup import OLMO_MODEL
+from langchain_core.callbacks import CallbackManager
 model_path = OLMO_MODEL
 # Environment variables