Red-tech-hub commited on
Commit
4bc1e98
1 Parent(s): a9122b2

[update] install added

Browse files
Files changed (3) hide show
  1. .env +1 -0
  2. requirements.txt +2 -1
  3. run.py +6 -0
.env ADDED
@@ -0,0 +1 @@
 
 
1
+ TRANSFORMERS_CACHE=/code/cache/
requirements.txt CHANGED
@@ -12,4 +12,5 @@ langchain_community==0.0.36
12
  langserve==0.1.1
13
  chromadb==0.4.24
14
  starlette==0.37.2
15
- typer==0.10.0
 
 
12
  langserve==0.1.1
13
  chromadb==0.4.24
14
  starlette==0.37.2
15
+ typer==0.10.0
16
+ sentence-transformers
run.py CHANGED
@@ -5,6 +5,12 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
5
  from langchain_community.vectorstores import Chroma
6
  from transformers import AutoModelForCausalLM
7
 
 
 
 
 
 
 
8
  model_kwargs = {'trust_remote_code': True}
9
 
10
  embedding = HuggingFaceEmbeddings(
 
5
  from langchain_community.vectorstores import Chroma
6
  from transformers import AutoModelForCausalLM
7
 
8
+ from dotenv import load_dotenv
9
+ load_dotenv()
10
+
11
+ import os
12
+ os.getenv("TRANSFORMERS_CACHE")
13
+
14
  model_kwargs = {'trust_remote_code': True}
15
 
16
  embedding = HuggingFaceEmbeddings(