Spaces:
Sleeping
Sleeping
Commit
·
1ff23c4
1
Parent(s):
de8b3d2
changed to german extractive model
Browse files- utils/config.py +3 -1
utils/config.py
CHANGED
|
@@ -13,7 +13,9 @@ parser.add_argument('--name', default="Document Insights: Extractive & Generativ
|
|
| 13 |
model_configs = {
|
| 14 |
'EMBEDDING_MODEL': os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L12-v2"),
|
| 15 |
'GENERATIVE_MODEL': os.getenv("GENERATIVE_MODEL", "gpt-4"),
|
| 16 |
-
'EXTRACTIVE_MODEL': os.getenv("EXTRACTIVE_MODEL", "deepset/roberta-base-squad2"),
|
|
|
|
|
|
|
| 17 |
'OPENAI_KEY': os.getenv("OPENAI_KEY"),
|
| 18 |
'COHERE_KEY': os.getenv("COHERE_KEY"),
|
| 19 |
}
|
|
|
|
| 13 |
model_configs = {
|
| 14 |
'EMBEDDING_MODEL': os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L12-v2"),
|
| 15 |
'GENERATIVE_MODEL': os.getenv("GENERATIVE_MODEL", "gpt-4"),
|
| 16 |
+
#'EXTRACTIVE_MODEL': os.getenv("EXTRACTIVE_MODEL", "deepset/roberta-base-squad2"),
|
| 17 |
+
'EXTRACTIVE_MODEL': os.getenv("EXTRACTIVE_MODEL", "deepset/gelectra-large-germanquad"),
|
| 18 |
+
#'EXTRACTIVE_MODEL': os.getenv("EXTRACTIVE_MODEL", "MachineLearningReply/bert-base-german-legal-qa"),
|
| 19 |
'OPENAI_KEY': os.getenv("OPENAI_KEY"),
|
| 20 |
'COHERE_KEY': os.getenv("COHERE_KEY"),
|
| 21 |
}
|