Update app.py

app.py

@@ -54,9 +54,11 @@ def remove_background(result):
     elif isinstance(result, str):
         result = Image.open(result)
         result = rembg.remove(result)
-    else:
-        # Convert the NumPy array to a PIL Image
-        result = Image.fromarray(result)
+    elif isinstance(result, numpy.ndarray):
+      print('here ELIF 2')
+      # Convert the NumPy array to a PIL Image
+      result = Image.fromarray(result)
+      result = rembg.remove(result)
     return result