deploy_4

app.py

@@ -381,6 +381,8 @@ def process_single_image(image_path, output_folder, bg_method, canvas_size_name,
             image_with_no_bg = remove_background_rembg(image_path)
         elif bg_method == 'bria':
             image_with_no_bg = remove_background_bria(image_path)
+        elif bg_method == None:
+            image_with_no_bg = Image.open(image_path)
         
         temp_image_path = os.path.join(output_folder, f"temp_{filename}")
         image_with_no_bg.save(temp_image_path, format='PNG')
@@ -500,7 +502,7 @@ def process_images(input_files, bg_method='rembg', watermark_path=None, canvas_s
                 avg_processing_time = (avg_processing_time * idx + image_processing_time) / (idx + 1)
                 if result:
                     if watermark_path:
-                        get_name = future_to_image[future].split('/')
+                        get_name = future_to_image[future].split('\\')
                         get_name = remove_extension(get_name[len(get_name)-1])
                         twibbon_input = f'{get_name}.png' if output_format == 'PNG' else f'{get_name}.jpg'
                         twibbon_output_path = os.path.join(output_folder, f'result_{start_time_image}.png')
@@ -644,7 +646,7 @@ with gr.Blocks(theme="NoCrypt/miku@1.2.2") as iface:
         num_workers = gr.Slider(minimum=1, maximum=16, step=1, label="Number of Workers", value=5)
 
     with gr.Row():
-        bg_method = gr.Radio(choices=["bria", "rembg"], label="Background Removal Method", value="bria")
+        bg_method = gr.Radio(choices=["bria", "rembg", None], label="Background Removal Method", value="bria")
         bg_choice = gr.Radio(choices=["transparent", "white", "custom"], label="Background Choice", value="white")
         custom_color = gr.ColorPicker(label="Custom Background Color", value="#ffffff", visible=False)