Spaces:
Sleeping
Sleeping
Lazar Radojevic
commited on
Commit
·
e7f4bff
1
Parent(s):
b3af506
add .cache for sentence transformers
Browse files- .cache/.gitkeep +0 -0
- run.py +4 -0
.cache/.gitkeep
ADDED
|
File without changes
|
run.py
CHANGED
|
@@ -3,6 +3,10 @@ from pydantic import BaseModel
|
|
| 3 |
from typing import List
|
| 4 |
from src.search_engine import PromptSearchEngine
|
| 5 |
from src.prompt_loader import PromptLoader
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
# Constants
|
| 8 |
SEED = 42
|
|
|
|
| 3 |
from typing import List
|
| 4 |
from src.search_engine import PromptSearchEngine
|
| 5 |
from src.prompt_loader import PromptLoader
|
| 6 |
+
import os
|
| 7 |
+
|
| 8 |
+
os.environ["SENTENCE_TRANSFORMERS_HOME"] = "./.cache"
|
| 9 |
+
|
| 10 |
|
| 11 |
# Constants
|
| 12 |
SEED = 42
|