at-engine / app /config.py
Godswill-IoT's picture
Upload 15 files
fb9f2a9 verified
"""
Configuration - Environment Variables Only
"""
import os
from typing import Optional
class Config:
"""Engine configuration from environment variables"""
# Hugging Face API
HF_TOKEN: str = os.getenv("HF_TOKEN", "")
# Model Configuration
HF_TEXT_MODEL: str = os.getenv("HF_TEXT_MODEL", "meta-llama/Meta-Llama-3-8B-Instruct")
HF_ASR_MODEL: str = os.getenv("HF_ASR_MODEL", "openai/whisper-base")
HF_TTS_MODEL: str = os.getenv("HF_TTS_MODEL", "facebook/fastspeech2-en-ljspeech")
HF_AVATAR_MODEL: str = os.getenv("HF_AVATAR_MODEL", "vinthony/SadTalker")
# API Configuration
ENGINE_NAME: str = "avatar-tutor-engine"
ENGINE_VERSION: str = "1.0.0"
# Server
HOST: str = os.getenv("HOST", "0.0.0.0")
PORT: int = int(os.getenv("PORT", "7860"))
@classmethod
def validate(cls) -> Optional[str]:
"""Validate required configuration"""
if not cls.HF_TOKEN:
return "HF_TOKEN environment variable is required"
return None
config = Config()