Ahmed-Alghamdi commited on
Commit
35588df
·
verified ·
1 Parent(s): 12d6a2d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +5 -11
config.py CHANGED
@@ -1,15 +1,9 @@
1
  # config.py
2
- import os
3
 
4
  class Config:
5
  DOCUMENT_FILE = 'RAG DATA.txt'
6
- EMBEDDING_MODEL = os.environ.get('EMBEDDING_MODEL', 'sentence-transformers/paraphrase-multilingual-mpnet-base-v2')
7
- LLM_MODEL = os.environ.get('LLM_MODEL', 'CAMeL-Lab/bert-base-arabic-camelbert-ca')
8
- BATCH_SIZE = int(os.environ.get('BATCH_SIZE', 32))
9
- TOP_K = int(os.environ.get('TOP_K', 3))
10
- MAX_LENGTH = int(os.environ.get('MAX_LENGTH', 1024))
11
- MIN_SIMILARITY_SCORE = float(os.environ.get('MIN_SIMILARITY_SCORE', 0.35))
12
- MAX_LENGTH = int(os.environ.get('MAX_LENGTH', 512))
13
- MAX_NEW_TOKENS = int(os.environ.get('MAX_NEW_TOKENS', 100))
14
-
15
-
 
1
  # config.py
 
2
 
3
  class Config:
4
  DOCUMENT_FILE = 'RAG DATA.txt'
5
+ EMBEDDING_MODEL = 'sentence-transformers/paraphrase-multilingual-mpnet-base-v2'
6
+ BATCH_SIZE = 32
7
+ TOP_K = 3
8
+ MIN_SIMILARITY_SCORE = 0.35
9
+ MAX_LENGTH = 512