import os | |
from dotenv import load_dotenv | |
# Load environment variables from .env file | |
load_dotenv() | |
# Hugging Face API token | |
HF_TOKEN = os.getenv("HF_TOKEN", "") | |
# Default model for text to image | |
DEFAULT_TEXT2IMG_MODEL = "stabilityai/stable-diffusion-2-1" | |
# Default model for image to image | |
DEFAULT_IMG2IMG_MODEL = "lllyasviel/sd-controlnet-depth" | |
# API settings | |
API_HOST = os.getenv("API_HOST", "0.0.0.0") | |
API_PORT = int(os.getenv("API_PORT", "8000")) | |
# Gradio settings | |
GRADIO_HOST = os.getenv("GRADIO_HOST", "0.0.0.0") | |
GRADIO_PORT = int(os.getenv("GRADIO_PORT", "7860")) | |