Spaces:
Running
Running
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -24,7 +24,7 @@ if uploaded_file:
|
|
| 24 |
if option == "Convert to Black & White":
|
| 25 |
result_img = convert_to_bw(image)
|
| 26 |
#Colorize a black and white image using a pre-trained model
|
| 27 |
-
elif option == "Colorize
|
| 28 |
with st.spinner("Colorizing..."):
|
| 29 |
net = load_colorization_model()
|
| 30 |
result_img = colorize_bw_image(image, net)
|
|
|
|
| 24 |
if option == "Convert to Black & White":
|
| 25 |
result_img = convert_to_bw(image)
|
| 26 |
#Colorize a black and white image using a pre-trained model
|
| 27 |
+
elif option == "Colorize Your B&W images":
|
| 28 |
with st.spinner("Colorizing..."):
|
| 29 |
net = load_colorization_model()
|
| 30 |
result_img = colorize_bw_image(image, net)
|