test v4

app.py

@@ -27,11 +27,20 @@ def fn(image):
 def process(image):
     def numpy_to_pil(image):
         """Convert a numpy array to a PIL Image."""
-        if image.dtype == np.uint8:  # Most common case
+        if not isinstance(image, np.ndarray):
+            raise TypeError("Input must be a numpy array")
+
+            # Determine the mode based on the shape and dtype of the image
+        if image.ndim == 2:  # Grayscale image
+            mode = "L"
+        elif image.ndim == 3 and image.shape[2] == 3:  # RGB image
             mode = "RGB"
+        elif image.ndim == 3 and image.shape[2] == 4:  # RGBA image
+            mode = "RGBA"
         else:
-            mode = "F"  # Floating point
-        return Image.fromarray(image.astype('uint8'), mode)
+            raise ValueError("Unsupported image shape: {}".format(image.shape))
+
+        return Image.fromarray(image, mode)
 
     image = numpy_to_pil(image)  # Convert numpy array to PIL Image
     buffered = BytesIO()