Arabic-Rag-Chatbot / config.py
Ahmed-Alghamdi's picture
Update config.py
74656f4 verified
raw
history blame
515 Bytes
# config.py
import os
class Config:
DOCUMENT_FILE = 'RAG DATA.txt'
EMBEDDING_MODEL = os.environ.get('EMBEDDING_MODEL', 'sentence-transformers/paraphrase-multilingual-mpnet-base-v2')
LLM_MODEL = os.environ.get('LLM_MODEL', 'CAMeL-Lab/bert-base-arabic-camelbert-ca')
BATCH_SIZE = int(os.environ.get('BATCH_SIZE', 32))
TOP_K = int(os.environ.get('TOP_K', 5))
MAX_LENGTH = int(os.environ.get('MAX_LENGTH', 1024))
MIN_SIMILARITY_SCORE = float(os.environ.get('MIN_SIMILARITY_SCORE', 0.35))