PankhuriSharma9795 commited on
Commit
4e4d80a
·
verified ·
1 Parent(s): 0ea9bf8

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -9
src/streamlit_app.py CHANGED
@@ -1,13 +1,6 @@
1
  import os
2
  os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
3
 
4
- os.environ["HF_HOME"] = "/app/cache"
5
- os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
6
- os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/app/cache"
7
- os.makedirs("/app/cache", exist_ok=True)
8
- os.environ["TRANSFORMERS_OFFLINE"] = "1"
9
- os.environ["HF_HUB_OFFLINE"] = "1"
10
-
11
 
12
  import pickle
13
  import numpy as np
@@ -49,8 +42,8 @@ except Exception as e:
49
  from sentence_transformers import SentenceTransformer
50
  from langchain_community.embeddings import HuggingFaceEmbeddings
51
 
52
- model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/app/cache")
53
- embed_model = HuggingFaceEmbeddings(model_name="./models/all-MiniLM-L6-v2")
54
 
55
 
56
  # ==================== Vector Store & Retriever ====================
 
1
  import os
2
  os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
3
 
 
 
 
 
 
 
 
4
 
5
  import pickle
6
  import numpy as np
 
42
  from sentence_transformers import SentenceTransformer
43
  from langchain_community.embeddings import HuggingFaceEmbeddings
44
 
45
+ model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
46
+ embed_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
47
 
48
 
49
  # ==================== Vector Store & Retriever ====================