try np fix when using other models from default

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from rembg import remove, new_session
 from PIL import Image
+import numpy as np
 import logging
 import time
 
@@ -14,7 +15,7 @@ MODEL_OPTIONS = {
     "isnet-general-use": "A new pre-trained model for general use cases",
     "isnet-anime": "High-accuracy segmentation for anime characters",
     "silueta": "A reduced-size version of u2net (43MB)",
-    "sam": "A pre-trained model for any use case",
+    "sam": "A pre-trained model for any use case (Segment Anything Model)",
     "unet": "Lightweight version of u2net model",
     "u2netp": "A lightweight version of u2net model",
     "u2net_human_seg": "A pre-trained model for human segmentation",
@@ -58,14 +59,21 @@ def remove_background(input_image, bg_color, model_choice, alpha_matting, post_p
         
         logging.info(f'Model name={model_name}')
         logging.info(remove_kwargs)
+
+        # Convert PIL Image to numpy array
+        input_array = np.array(input_image)
+
         # Use the remove function
         if session or bg_color_rgba:
-            output_image = remove(input_image, **{k: v for k, v in remove_kwargs.items() if v is not None})
+            output_array = remove(input_array, **{k: v for k, v in remove_kwargs.items() if v is not None})
         else:
-            output_image = remove(input_image)  # Use the default remove function
+            output_array = remove(input_array)  # Use the default remove function
 
         logging.info("Background removed")
 
+        # Convert numpy array back to PIL Image
+        output_image = Image.fromarray(output_array)
+
         # Convert to RGB mode if necessary
         if not only_mask and output_image.mode != 'RGB':
             output_image = output_image.convert('RGB')
@@ -74,7 +82,7 @@ def remove_background(input_image, bg_color, model_choice, alpha_matting, post_p
         # Save the output image to a temporary file
         # Generate a unique timestamp for the output file name
         timestamp = time.strftime("%Y%m%d-%H%M%S")
-        output_path = f"output_remove_background_{timestamp}.png"
+        output_path = f"output_remove_background_s{timestamp}.png"
         output_image.save(output_path)
         logging.info(f"Saved output image {output_path}")
 
@@ -104,7 +112,7 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Output Image"), 
         gr.File(label="Download the output image")
     ],
-    title="Advanced Background Remover v2.5",
+    title="Advanced Background Remover v2.6",
     description="Upload an image to remove the background. Customize the result with different options, including background color, model selection, alpha matting, and more.",
     allow_flagging="never",
 )