Spaces:
Runtime error
Runtime error
Update src/streamlit_app.py
Browse files- 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"
|
| 53 |
-
embed_model = HuggingFaceEmbeddings(model_name="
|
| 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 ====================
|