Upload folder using huggingface_hub

app.py

@@ -81,7 +81,7 @@ def run(content_image, style_name, style_strength=5):
             future_all = executor.submit(run_inference, False)
             future_bg = executor.submit(run_inference, True)
         generated_img_all, _ = future_all.result()
-        generated_img_bg, bg_ratio = future_bg.result()
+        generated_img_bg, salient_object_ratio = future_bg.result()
 
     et = time.time()
     print('TIME TAKEN:', et-st)
@@ -89,7 +89,7 @@ def run(content_image, style_name, style_strength=5):
     yield (
         (content_image, postprocess_img(generated_img_all, original_size)),
         (content_image, postprocess_img(generated_img_bg, original_size)),
-        f'{bg_ratio:.2f}'
+        f'{salient_object_ratio:.2f}'
     )
 
 def set_slider(value):
@@ -126,7 +126,7 @@ with gr.Blocks(css=css) as demo:
             download_button_1 = gr.DownloadButton(label='Download Styled Image', visible=False)
             with gr.Group():
                 output_image_background = ImageSlider(position=0.15, label='Styled Background', type='pil', interactive=False, show_download_button=False)
-                bg_ratio_label = gr.Label(label='Background Ratio')
+                salient_object_ratio_label = gr.Label(label='Salient Object Ratio')
             download_button_2 = gr.DownloadButton(label='Download Styled Background', visible=False)
 
     def save_image(img_tuple1, img_tuple2):
@@ -143,7 +143,7 @@ with gr.Blocks(css=css) as demo:
     submit_button.click(
         fn=run, 
         inputs=[content_image, style_dropdown, style_strength_slider], 
-        outputs=[output_image_all, output_image_background, bg_ratio_label]
+        outputs=[output_image_all, output_image_background, salient_object_ratio_label]
     ).then(
         fn=save_image,
         inputs=[output_image_all, output_image_background],

inference.py

@@ -60,16 +60,16 @@ def inference(
         content_features = model(content_image)
 
         resized_bg_masks = []
-        background_ratio = None
+        salient_object_ratio = None
         if apply_to_background:            
             segmentation_output = segmentation_model(content_image)['out']
             segmentation_mask = segmentation_output.argmax(dim=1)
             background_mask = (segmentation_mask == 0).float()
             foreground_mask = 1 - background_mask
             
-            background_pixel_count = background_mask.sum().item()
+            salient_object_pixel_count = foreground_mask.sum().item()
             total_pixel_count = segmentation_mask.numel()
-            background_ratio = background_pixel_count / total_pixel_count
+            salient_object_ratio = salient_object_pixel_count / total_pixel_count
 
             for cf in content_features:
                 _, _, h_i, w_i = cf.shape
@@ -106,4 +106,4 @@ def inference(
     if DEV_MODE:
         writer.flush()
         writer.close()
-    return generated_image, background_ratio
+    return generated_image, salient_object_ratio