Uzaiir commited on
Commit
c50decd
·
verified ·
1 Parent(s): ff2246c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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 Black & White":
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)