Update app.py

app.py

@@ -51,7 +51,10 @@ def find_similarity(base64_image, text_input):
 
 # Define a function for image segmentation
 def segment_image(input_image, text_input):
-    image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)
+    image_bytes = base64.b64decode(input_image)
+    image = Image.open(BytesIO(image_bytes))
+    
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     mask_generator = SamAutomaticMaskGenerator(sam)
     masks = mask_generator.generate(image)
 
@@ -81,7 +84,7 @@ def segment_image(input_image, text_input):
     return segmented_regions
 
 # Create Gradio components
-input_image = gr.inputs.Image()
+input_image = gr.Textbox(label="Base64 Image", lines=8)
 text_input = gr.Textbox(label="Text Input")  # Use Textbox with a label
 output_images = gr.outputs.JSON()