CSB261 commited on
Commit
d59a3eb
ยท
verified ยท
1 Parent(s): 94a16e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -1,7 +1,11 @@
1
  import gradio as gr
2
  from PIL import Image, ImageEnhance, ImageOps, ImageFilter
 
3
 
4
  def convert_to_moody_bw(image):
 
 
 
5
  # ์ด๋ฏธ์ง€๋ฅผ ํ‘๋ฐฑ์œผ๋กœ ๋ณ€ํ™˜
6
  bw_image = ImageOps.grayscale(image)
7
 
@@ -19,10 +23,20 @@ def convert_to_moody_bw(image):
19
  return bw_image
20
 
21
  def convert_and_prepare_download(image):
 
 
 
22
  bw_image = convert_to_moody_bw(image)
 
23
  # JPG ํ˜•์‹์œผ๋กœ ๋ณ€ํ™˜
24
  jpg_image = bw_image.convert("RGB")
25
- return bw_image, jpg_image
 
 
 
 
 
 
26
 
27
  with gr.Blocks() as demo:
28
  gr.Markdown("## ๋ถ„์œ„๊ธฐ ์žˆ๋Š” ํ‘๋ฐฑ ์‚ฌ์ง„ ๋ณ€ํ™˜๊ธฐ")
@@ -41,4 +55,4 @@ with gr.Blocks() as demo:
41
  outputs=[output_image, download_link]
42
  )
43
 
44
- demo.launch()
 
1
  import gradio as gr
2
  from PIL import Image, ImageEnhance, ImageOps, ImageFilter
3
+ import io
4
 
5
  def convert_to_moody_bw(image):
6
+ """
7
+ ์ด๋ฏธ์ง€๋ฅผ ํ‘๋ฐฑ์œผ๋กœ ๋ณ€ํ™˜ํ•˜๊ณ  ๋ถ„์œ„๊ธฐ๋ฅผ ๋”ํ•ฉ๋‹ˆ๋‹ค.
8
+ """
9
  # ์ด๋ฏธ์ง€๋ฅผ ํ‘๋ฐฑ์œผ๋กœ ๋ณ€ํ™˜
10
  bw_image = ImageOps.grayscale(image)
11
 
 
23
  return bw_image
24
 
25
  def convert_and_prepare_download(image):
26
+ """
27
+ ์ด๋ฏธ์ง€๋ฅผ ํ‘๋ฐฑ์œผ๋กœ ๋ณ€ํ™˜ํ•˜๊ณ , JPG ํ˜•์‹์œผ๋กœ ๋ณ€ํ™˜ํ•˜์—ฌ ๋‹ค์šด๋กœ๋“œํ•  ์ˆ˜ ์žˆ๋„๋ก ์ค€๋น„ํ•ฉ๋‹ˆ๋‹ค.
28
+ """
29
  bw_image = convert_to_moody_bw(image)
30
+
31
  # JPG ํ˜•์‹์œผ๋กœ ๋ณ€ํ™˜
32
  jpg_image = bw_image.convert("RGB")
33
+
34
+ # BytesIO ๊ฐ์ฒด์— ์ด๋ฏธ์ง€ ์ €์žฅ
35
+ buf = io.BytesIO()
36
+ jpg_image.save(buf, format="JPEG")
37
+ buf.seek(0)
38
+
39
+ return bw_image, buf
40
 
41
  with gr.Blocks() as demo:
42
  gr.Markdown("## ๋ถ„์œ„๊ธฐ ์žˆ๋Š” ํ‘๋ฐฑ ์‚ฌ์ง„ ๋ณ€ํ™˜๊ธฐ")
 
55
  outputs=[output_image, download_link]
56
  )
57
 
58
+ demo.launch()