ssboost commited on
Commit
29f2cfb
·
verified ·
1 Parent(s): c6538dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -8
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
- CLARITY_MODEL = os.environ.get("CLARITY_MODEL", "")
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 CLARITY_MODEL:
25
- logger.warning("CLARITY_MODEL이 설정되지 않았습니다. 환경 변수를 설정해주세요.")
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 CLARITY_MODEL:
68
- raise ValueError("CLARITY_MODEL이 설정되지 않았습니다.")
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
- CLARITY_MODEL,
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"),