WebProject / backend /config.py
Soheib31's picture
Upload 35 files
f8c8688 verified
raw
history blame contribute delete
533 Bytes
import os
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
# Get Hugging Face API token
HUGGINGFACE_API_TOKEN = os.getenv('HUGGINGFACE_API_TOKEN')
# API configuration
API_CONFIG = {
'token': HUGGINGFACE_API_TOKEN,
'models': {
'summarization': 'meta-llama/Llama-2-70b-chat-hf',
'question_answering': 'deepseek-ai/DeepSeek-V2-Chat',
'image_captioning': 'Qwen/Qwen2.5-VL-7B-Instruct',
'translation': 'facebook/mbart-large-50-many-to-many-mmt'
}
}