Spaces:
Sleeping
Sleeping
Sam-Oliveira
commited on
Commit
·
0fb074a
1
Parent(s):
8458d3b
fix DB connection:
Browse files- src/config.py +3 -2
src/config.py
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
from pathlib import Path
|
| 2 |
-
|
| 3 |
# Root folder for DB
|
| 4 |
-
PROJ = Path(__file__).parent
|
|
|
|
| 5 |
MAX_RESULTS = 10 #default number of results
|
| 6 |
|
| 7 |
MODEL_NAME = "unsloth/llama-3-8b-Instruct-bnb-4bit" #default model
|
|
|
|
| 1 |
from pathlib import Path
|
| 2 |
+
import pathlib,tempfile
|
| 3 |
# Root folder for DB
|
| 4 |
+
#PROJ = Path(__file__).parent # For MAC
|
| 5 |
+
PROJ = pathlib.Path(tempfile.gettempdir()) # For Space
|
| 6 |
MAX_RESULTS = 10 #default number of results
|
| 7 |
|
| 8 |
MODEL_NAME = "unsloth/llama-3-8b-Instruct-bnb-4bit" #default model
|