arxivgpt kim commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -132,7 +132,7 @@ if colorizer is not None:
|
|
| 132 |
with st.expander("Original photo", True):
|
| 133 |
st.image(img_input)
|
| 134 |
|
| 135 |
-
if st.button("
|
| 136 |
|
| 137 |
with st.spinner("AI is doing the magic!"):
|
| 138 |
img_output = colorize_image(img_input)
|
|
@@ -144,7 +144,7 @@ if colorizer is not None:
|
|
| 144 |
img_input.convert("RGB").save(f"./output/{now}-input.jpg")
|
| 145 |
img_output.convert("RGB").save(f"./output/{now}-output.jpg")
|
| 146 |
|
| 147 |
-
st.write("
|
| 148 |
st.image(img_output)
|
| 149 |
# reuse = st.button('Edit again (Re-use this image)', on_click=set_image, args=(inpainted_img, ))
|
| 150 |
|
|
|
|
| 132 |
with st.expander("Original photo", True):
|
| 133 |
st.image(img_input)
|
| 134 |
|
| 135 |
+
if st.button("컬러 복원하기") and uploaded_file is not None:
|
| 136 |
|
| 137 |
with st.spinner("AI is doing the magic!"):
|
| 138 |
img_output = colorize_image(img_input)
|
|
|
|
| 144 |
img_input.convert("RGB").save(f"./output/{now}-input.jpg")
|
| 145 |
img_output.convert("RGB").save(f"./output/{now}-output.jpg")
|
| 146 |
|
| 147 |
+
st.write("컬러 복원중입니다. 잠시만 기다리세요.")
|
| 148 |
st.image(img_output)
|
| 149 |
# reuse = st.button('Edit again (Re-use this image)', on_click=set_image, args=(inpainted_img, ))
|
| 150 |
|