|
a |
|
b/modules/chatbot/config.py |
|
|
1 |
class Config: |
|
|
2 |
chat_params = { |
|
|
3 |
"gpt_tok": "danielpark/medical-QA-chatGPT2-tok-v1", |
|
|
4 |
"tf_gpt_model": "danielpark/medical-QA-chatGPT2-v1", |
|
|
5 |
"bert_tok": "danielpark/medical-QA-BioRedditBERT-uncased-v1", |
|
|
6 |
"tf_q_extractor": "question_extractor_model", |
|
|
7 |
"data": "danielpark/MQuAD-v1", |
|
|
8 |
"max_answer_len": 20, |
|
|
9 |
"isEval": False, |
|
|
10 |
"runDocker": True, # Exceeds the bandwidth of git-lfs, mounts to local storage to find folder location for free use. I use the python utifunction package. |
|
|
11 |
"container_mounted_folder_path": "/usr/src/app/data", |
|
|
12 |
} |