Spaces:
Sleeping
Sleeping
File size: 615 Bytes
55fd638 | 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 | 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() |