Spaces:
Paused
Paused
Tao Wu commited on
Commit ·
babde94
1
Parent(s): 3310e88
fix bug
Browse files- app/config.py +1 -1
- app/embedding_setup.py +1 -1
app/config.py
CHANGED
|
@@ -16,7 +16,7 @@ ENCODE_KWARGS = {
|
|
| 16 |
QUERY_INSTRUCTION = os.getenv('QUERY_INSTRUCTION', '')
|
| 17 |
|
| 18 |
# Other configurations
|
| 19 |
-
TOP_K = int(os.getenv('TOP_K',
|
| 20 |
#PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/course_emb_db")
|
| 21 |
PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/BA_course")
|
| 22 |
CSV_FILE_PATH = os.getenv('CSV_FILE_PATH', '/app/data/occupations_de.csv')
|
|
|
|
| 16 |
QUERY_INSTRUCTION = os.getenv('QUERY_INSTRUCTION', '')
|
| 17 |
|
| 18 |
# Other configurations
|
| 19 |
+
TOP_K = int(os.getenv('TOP_K', 10))
|
| 20 |
#PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/course_emb_db")
|
| 21 |
PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/BA_course")
|
| 22 |
CSV_FILE_PATH = os.getenv('CSV_FILE_PATH', '/app/data/occupations_de.csv')
|
app/embedding_setup.py
CHANGED
|
@@ -31,7 +31,7 @@ retriever = db.as_retriever(search_kwargs={"k": TOP_K})
|
|
| 31 |
|
| 32 |
|
| 33 |
LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
|
| 34 |
-
lora_weights = "/
|
| 35 |
|
| 36 |
hf_auth = os.environ.get("hf_token")
|
| 37 |
|
|
|
|
| 31 |
|
| 32 |
|
| 33 |
LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
|
| 34 |
+
lora_weights = "wt3639/Llama-3-8B-Instruct_CourseRec_lora"
|
| 35 |
|
| 36 |
hf_auth = os.environ.get("hf_token")
|
| 37 |
|