Spaces:
Running
Running
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' | |
} | |
} |