added progress bar

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import uuid
 from zipfile import ZipFile, ZIP_DEFLATED
 from io import BytesIO
+from stqdm import stqdm
 
 # Import util functions from deoldify
 # NOTE:  This must be the first call in order to work properly!
@@ -124,12 +125,14 @@ def display_single_image(uploaded_file, img_size=800):
     st_title_message.markdown("**To begin, please upload an image** 👇")
 
 def process_multiple_images(uploaded_files, img_size=800):
+
     num_imgs = len(uploaded_files)
 
     output_images_list = []
     img_names_list = []
     idx = 1
-    for idx, uploaded_file in enumerate(uploaded_files, start=1):
+
+    for idx, uploaded_file in stqdm(enumerate(uploaded_files, start=1)):
         st_title_message.markdown("**Processing image {}/{}. Please wait** ⌛".format(idx,
                                                                                     num_imgs))
 
@@ -231,6 +234,7 @@ if colorizer is not None:
     st_title_message.markdown("**To begin, please upload an image** 👇")
 
     #Choose your own image
+    uploaded_files = []
     uploaded_files = st_file_uploader.file_uploader("Upload a black and white photo", 
                                                     type=['png', 'jpg', 'jpeg'],
                                                     accept_multiple_files=True)