Update app.py

app.py

@@ -6,6 +6,7 @@ from rembg import remove
 import gradio as gr
 import spaces
 import io
+import numpy as np
 
 # Load the Nomic embed model
 processor = AutoImageProcessor.from_pretrained("nomic-ai/nomic-embed-vision-v1.5")
@@ -21,6 +22,9 @@ def focus_on_subject(image: Image.Image) -> Image.Image:
     Returns:
         PIL.Image.Image: Cropped image with background removed.
     """
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+        
     image = image.convert("RGB")
     
     # Remove background