comic-grading / parameters.py
Manish Gupta
Modified codebase to make the loading times drastically better.
e3460b7
raw
history blame
538 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
AWS_BUCKET = os.getenv("AWS_BUCKET")
os.environ["AWS_ACCESS_KEY_ID"] = os.getenv("AWS_ACCESS_KEY_ID")
os.environ["AWS_SECRET_ACCESS_KEY"] = os.getenv("AWS_SECRET_ACCESS_KEY")
os.environ["S3_BUCKET_NAME"] = os.getenv("AWS_BUCKET")
VISUAL_CHOICES = ["DARK", "FLUX_COMIC", "GHIBLI_COMIC"]
MAX_TRIES = os.getenv("MAX_TRIES")
OPEN_AI_API_KEY = os.getenv("OPEN_AI_KEY")
MODEL_SERVER_URL = os.getenv("MODEL_SERVER_URL")
IMG_HEIGHT=os.getenv("IMG_HEIGHT")
IMG_WIDTH=os.getenv("IMG_WIDTH")