Enhance image processing in remove_background function by adding dtype check and conversion for extracted images. This ensures that non-uint8 images are properly scaled and converted before being transformed into PIL format.

app.py

@@ -76,6 +76,8 @@ def remove_background(
         extracted_img, mask = dis_remove_background(DIS_ONNX_MODEL_PATH, temp.name)
     # Convert to PIL Image if needed
     if isinstance(extracted_img, np.ndarray):
+        if extracted_img.dtype != np.uint8:
+            extracted_img = (np.clip(extracted_img, 0, 1) * 255).astype(np.uint8)
         extracted_img = Image.fromarray(extracted_img)
     return extracted_img