Spaces:
Running
Running
File size: 1,221 Bytes
038d6db a8f90b0 9e38f34 a8f90b0 9e38f34 a8f90b0 9e38f34 a8f90b0 9e38f34 a8f90b0 038d6db 9e38f34 081369c a8f90b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
MODEL_GEMINI = "gemini-3-pro-preview"
MODEL_OLMOCR = "olmOCR-2-7B-1025-FP8"
MODELS_MAP = {
# GPT-5.2 family
"gpt-5.2": {"input": 1.75, "output": 14.00, "backend": "openai"},
"gpt-5.2-chat-latest": {"input": 1.75, "output": 14.00, "backend": "openai"},
"gpt-5.2-pro": {"input": 21.00, "output": 168.00, "backend": "openai"},
# GPT-5.1 / GPT-5 family
"gpt-5.1": {"input": 1.25, "output": 10.00, "backend": "openai"},
"gpt-5": {"input": 1.25, "output": 10.00, "backend": "openai"},
"gpt-5-mini": {"input": 0.25, "output": 2.00, "backend": "openai"},
"gpt-5-nano": {"input": 0.05, "output": 0.40, "backend": "openai"},
# GPT-4.1 family
"gpt-4.1": {"input": 2.00, "output": 8.00, "backend": "openai"},
"gpt-4.1-mini": {"input": 0.40, "output": 1.60, "backend": "openai"},
"gpt-4.1-nano": {"input": 0.10, "output": 0.40, "backend": "openai"},
# Other backends (mock rates)
MODEL_GEMINI: {"input": 2.00, "output": 12.00, "backend": "gemini"},
MODEL_OLMOCR: {"input": 1.35, "output": 0.30, "backend": "olmocr"},
"gemini-2.5-pro": {"input": 1.25, "output": 10.00, "backend": "gemini"},
"default": {"input": 2.50, "output": 10.00, "backend": "openai"},
}
|