fixed minor issues

app.py

@@ -130,7 +130,6 @@ def process_multiple_images(uploaded_files, img_size=800):
     img_names_list = []
     idx = 1
     for idx, uploaded_file in enumerate(uploaded_files, start=1):
-        
         st_title_message.markdown("**Processing image {}/{}. Please wait** ⌛".format(idx,
                                                                                     num_imgs))
 
@@ -144,8 +143,7 @@ def process_multiple_images(uploaded_files, img_size=800):
         resized_pil_img = PIL.Image.fromarray(resized_img_rgb)
 
         # Send the image to the model
-        # output_pil_img = colorizer.plot_transformed_pil_image(resized_pil_img, render_factor=35, compare=False)
-        output_pil_img = resized_pil_img
+        output_pil_img = colorizer.plot_transformed_pil_image(resized_pil_img, render_factor=35, compare=False)
 
         output_images_list.append(output_pil_img)
         img_names_list.append(img_name.split('.')[0])
@@ -218,15 +216,14 @@ st_color_option = st.sidebar.selectbox('Select colorizer mode',
 # det_nms_thres = st.sidebar.slider("Non-maximum supression IoU", 0.1, 0.9, value=0.4, step=0.1)
 
 # Load models
-print('before loading the model')
 try:
+    print('before loading the model')
     colorizer = load_model(model_folder, st_color_option)
-except:
-    pass
-    colorizer = True
-    # print('Error while loading the model. Please refresh the page')
-print('after loading the model')
+    print('after loading the model')
 
+except:
+    colorizer = None
+    print('Error while loading the model. Please refresh the page')
 
 if colorizer is not None:
     st_title_message.markdown("**To begin, please upload an image** 👇")