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"),
|