sourabhzanwar commited on
Commit
ee0b8dd
1 Parent(s): 3c91ce2

Update utils/config.py

Browse files
Files changed (1) hide show
  1. utils/config.py +2 -2
utils/config.py CHANGED
@@ -13,8 +13,8 @@ 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
- #'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"),
 
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"),