historical-ocr / backup /config.py
milwright's picture
submit pull for merge
85bdb4e verified
raw
history blame
645 Bytes
# config.py
"""
Configuration file for Mistral OCR processing.
Contains API key and other settings.
"""
import os
# Your Mistral API key - get from Hugging Face secrets or environment variable
# The priority order is: HF_SPACES environment var > regular environment var > empty string
# Note: No default API key is provided for security reasons
MISTRAL_API_KEY = os.environ.get("HF_MISTRAL_API_KEY", # First check HF-specific env var
os.environ.get("MISTRAL_API_KEY", "")) # Then check regular env var
# Model settings
OCR_MODEL = "mistral-ocr-latest"
TEXT_MODEL = "ministral-8b-latest"
VISION_MODEL = "pixtral-12b-latest"