Update app.py

app.py

@@ -7,7 +7,10 @@ def upscale_image(input_image, radio_input):
     upscale_factor = int(radio_input)  # 업스케일 비율을 정수형으로 변환
     # 입력 이미지를 주어진 업스케일 비율로 크기 변경 (보간법: INTER_CUBIC)
     output_image = cv2.resize(input_image, None, fx=upscale_factor, fy=upscale_factor, interpolation=cv2.INTER_CUBIC)
-    return output_image
+    # 업스케일된 이미지를 PNG 파일로 저장
+    output_file_path = "/mnt/data/upscaled_image.png"
+    cv2.imwrite(output_file_path, output_image)
+    return output_image, output_file_path
 
 # 인터페이스 설명
 DESCRIPTION = """
@@ -15,13 +18,13 @@ DESCRIPTION = """
 """
 
 # 업스케일 레벨 선택 라디오 버튼
-radio_input = gr.Radio(label="업스케일 레벨 선택 (Select Upscaling level)", choices=[2, 4, 6, 8, 10], value=2)
+radio_input = gr.Radio(label="업스케일 레벨 선택 (Select Upscaling level)", choices=[2, 4, 6, 8], value=2)
 
 # Gradio 인터페이스 정의
 demo = gr.Interface(
     fn=upscale_image,
     inputs=[gr.Image(label="입력 이미지 (Input Image)", type="numpy"), radio_input],
-    outputs=gr.Image(label="업스케일된 이미지(Upscaled Image)"),
+    outputs=[gr.Image(label="업스케일된 이미지(Upscaled Image)"), gr.File(label="PNG 파일 다운로드 (Download PNG File)")],
     title="Image Upscaler",
     description=DESCRIPTION
 )