Update pages/03_🖼️_Input_Images.py

pages/03_🖼️_Input_Images.py

@@ -68,12 +68,12 @@ def main():
                         with col2:
                             with st.spinner("Colorizing image..."):
                                 out_img, new_img = colorize_image(file, loaded_model)
-                                new_img.save("IMG_" + str(i) + ".jpg")
+                                new_img.save("IMG_" + str(i+1) + ".jpg")
                                 st.image(out_img, use_column_width="always")
 
                     else:
                         out_img, new_img = colorize_image(file, loaded_model)
-                        new_img.save("IMG_" + str(i + 1) + ".jpg")
+                        new_img.save("IMG_" + str(i+1) + ".jpg")
 
             if len(uploaded_file) > 1:
                 # Create a zip file