imagencpu / config.py
Abe
initial copy
8247a04
raw
history blame
578 Bytes
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"))