Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ logger = logging.getLogger(__name__)
|
|
| 12 |
# 환경변수에서 설정값들 가져오기
|
| 13 |
REPLICATE_API_TOKEN = os.environ.get("REPLICATE_API_TOKEN", "")
|
| 14 |
APP_PASSWORD = os.environ.get("APP_PASSWORD", "")
|
| 15 |
-
|
| 16 |
|
| 17 |
# 환경변수 검증
|
| 18 |
if not REPLICATE_API_TOKEN:
|
|
@@ -21,8 +21,8 @@ if not REPLICATE_API_TOKEN:
|
|
| 21 |
if not APP_PASSWORD:
|
| 22 |
logger.warning("APP_PASSWORD가 설정되지 않았습니다. 환경 변수를 설정해주세요.")
|
| 23 |
|
| 24 |
-
if not
|
| 25 |
-
logger.warning("
|
| 26 |
|
| 27 |
# 임시 디렉토리 생성
|
| 28 |
TEMP_DIR = os.path.join(tempfile.gettempdir(), "image_processor")
|
|
@@ -64,8 +64,8 @@ def upscale_image(
|
|
| 64 |
if not REPLICATE_API_TOKEN:
|
| 65 |
raise ValueError("REPLICATE_API_TOKEN이 설정되지 않았습니다.")
|
| 66 |
|
| 67 |
-
if not
|
| 68 |
-
raise ValueError("
|
| 69 |
|
| 70 |
try:
|
| 71 |
# 이미지 저장
|
|
@@ -77,8 +77,6 @@ def upscale_image(
|
|
| 77 |
replicate.client.Client(api_token=REPLICATE_API_TOKEN)
|
| 78 |
|
| 79 |
# 로그 정보 출력
|
| 80 |
-
logger.info("=== Clarity Upscaler 실행 정보 ===")
|
| 81 |
-
logger.info(f"모델: {CLARITY_MODEL}")
|
| 82 |
logger.info(f"SD 모델: {sd_model}")
|
| 83 |
logger.info(f"스케일 팩터: {scale_factor}")
|
| 84 |
logger.info(f"출력 형식: {output_format}")
|
|
@@ -90,7 +88,7 @@ def upscale_image(
|
|
| 90 |
|
| 91 |
# 환경변수에서 가져온 모델로 API 호출
|
| 92 |
output = replicate.run(
|
| 93 |
-
|
| 94 |
input={
|
| 95 |
"seed": seed,
|
| 96 |
"image": open(temp_input_path, "rb"),
|
|
|
|
| 12 |
# 환경변수에서 설정값들 가져오기
|
| 13 |
REPLICATE_API_TOKEN = os.environ.get("REPLICATE_API_TOKEN", "")
|
| 14 |
APP_PASSWORD = os.environ.get("APP_PASSWORD", "")
|
| 15 |
+
REPLICATE_MODEL = os.environ.get("REPLICATE_MODEL", "")
|
| 16 |
|
| 17 |
# 환경변수 검증
|
| 18 |
if not REPLICATE_API_TOKEN:
|
|
|
|
| 21 |
if not APP_PASSWORD:
|
| 22 |
logger.warning("APP_PASSWORD가 설정되지 않았습니다. 환경 변수를 설정해주세요.")
|
| 23 |
|
| 24 |
+
if not REPLICATE_MODEL:
|
| 25 |
+
logger.warning("REPLICATE_MODEL이 설정되지 않았습니다. 환경 변수를 설정해주세요.")
|
| 26 |
|
| 27 |
# 임시 디렉토리 생성
|
| 28 |
TEMP_DIR = os.path.join(tempfile.gettempdir(), "image_processor")
|
|
|
|
| 64 |
if not REPLICATE_API_TOKEN:
|
| 65 |
raise ValueError("REPLICATE_API_TOKEN이 설정되지 않았습니다.")
|
| 66 |
|
| 67 |
+
if not REPLICATE_MODEL:
|
| 68 |
+
raise ValueError("REPLICATE_MODEL이 설정되지 않았습니다.")
|
| 69 |
|
| 70 |
try:
|
| 71 |
# 이미지 저장
|
|
|
|
| 77 |
replicate.client.Client(api_token=REPLICATE_API_TOKEN)
|
| 78 |
|
| 79 |
# 로그 정보 출력
|
|
|
|
|
|
|
| 80 |
logger.info(f"SD 모델: {sd_model}")
|
| 81 |
logger.info(f"스케일 팩터: {scale_factor}")
|
| 82 |
logger.info(f"출력 형식: {output_format}")
|
|
|
|
| 88 |
|
| 89 |
# 환경변수에서 가져온 모델로 API 호출
|
| 90 |
output = replicate.run(
|
| 91 |
+
REPLICATE_MODEL,
|
| 92 |
input={
|
| 93 |
"seed": seed,
|
| 94 |
"image": open(temp_input_path, "rb"),
|