Update app.py

app.py

@@ -27,7 +27,7 @@ def process_image(image, prompt):
     return mask
 
 def get_masks(prompts, img, threshold):
-    prompts = [p.strip() for p in prompts.split(",")]
+    prompts = prompts.split(",")
     masks = []
     for prompt in prompts:
         mask = process_image(img, prompt)
@@ -43,7 +43,7 @@ def extract_image(pos_prompts, neg_prompts, img, threshold):
     neg_mask = np.any(np.stack(negative_masks), axis=0)
     final_mask = pos_mask & ~neg_mask
     
-    final_mask = Image.fromarray((final_mask * 255).astype(np.uint8), "L")
+    final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
     output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
     output_image.paste(img, mask=final_mask)
     return output_image, final_mask
@@ -62,7 +62,10 @@ iface = gr.Interface(
     ],
 )
 
-# Define an API endpoint
+# Launch Gradio UI
+iface.launch()
+
+# Define API interface
 api_interface = gr.Interface(
     fn=extract_image,
     inputs=[
@@ -78,6 +81,5 @@ api_interface = gr.Interface(
     live=True  # Setting live to True enables the API endpoint
 )
 
-# Run the Gradio UI and API
-iface.launch()
+# Launch API
 api_interface.launch(share=True)  # share=True allows external access to the API