Spaces:
Running
Running
| from pydantic_settings import BaseSettings | |
| from typing import Optional | |
| class Settings(BaseSettings): | |
| # Model settings | |
| MODEL_PATH: str = "./models/smartbugs-expert" | |
| HF_MODEL_ID: Optional[str] = None # e.g. "your-hf-username/smart-contract-auditor" | |
| HF_TOKEN: Optional[str] = None | |
| USE_PATTERN_FALLBACK: bool = True | |
| # Generation settings | |
| MAX_NEW_TOKENS: int = 300 | |
| TEMPERATURE: float = 0.7 | |
| MAX_INPUT_LENGTH: int = 1024 | |
| # Server | |
| HOST: str = "0.0.0.0" | |
| PORT: int = 8000 | |
| class Config: | |
| env_file = ".env" | |
| settings = Settings() |