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 |
|