jpjp9292 commited on
Commit
164755a
โ€ข
1 Parent(s): f6f13ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ def upscale_image(input_image, radio_input):
10
  # ์—…์Šค์ผ€์ผ๋œ ์ด๋ฏธ์ง€๋ฅผ PNG ํŒŒ์ผ๋กœ ์ €์žฅ
11
  output_file_path = "/mnt/data/upscaled_image.png"
12
  cv2.imwrite(output_file_path, output_image)
13
- return output_image, output_file_path
14
 
15
  # ์ธํ„ฐํŽ˜์ด์Šค ์„ค๋ช…
16
  DESCRIPTION = """
@@ -24,7 +24,7 @@ radio_input = gr.Radio(label="์—…์Šค์ผ€์ผ ๋ ˆ๋ฒจ ์„ ํƒ (Select Upscaling level
24
  demo = gr.Interface(
25
  fn=upscale_image,
26
  inputs=[gr.Image(label="์ž…๋ ฅ ์ด๋ฏธ์ง€ (Input Image)", type="numpy"), radio_input],
27
- outputs=[gr.Image(label="์—…์Šค์ผ€์ผ๋œ ์ด๋ฏธ์ง€(Upscaled Image)"), gr.File(label="PNG ํŒŒ์ผ ๋‹ค์šด๋กœ๋“œ (Download PNG File)")],
28
  title="Image Upscaler",
29
  description=DESCRIPTION
30
  )
 
10
  # ์—…์Šค์ผ€์ผ๋œ ์ด๋ฏธ์ง€๋ฅผ PNG ํŒŒ์ผ๋กœ ์ €์žฅ
11
  output_file_path = "/mnt/data/upscaled_image.png"
12
  cv2.imwrite(output_file_path, output_image)
13
+ return output_file_path
14
 
15
  # ์ธํ„ฐํŽ˜์ด์Šค ์„ค๋ช…
16
  DESCRIPTION = """
 
24
  demo = gr.Interface(
25
  fn=upscale_image,
26
  inputs=[gr.Image(label="์ž…๋ ฅ ์ด๋ฏธ์ง€ (Input Image)", type="numpy"), radio_input],
27
+ outputs=gr.File(label="PNG ํŒŒ์ผ ๋‹ค์šด๋กœ๋“œ (Download PNG File)"),
28
  title="Image Upscaler",
29
  description=DESCRIPTION
30
  )