Update app.py

app.py

@@ -89,6 +89,9 @@ def segment_image(input_image, text_input):
     # Sort the segmented images by similarity in descending order
     segmented_regions.sort(key=lambda x: x["similarity"], reverse=True)
 
+    # Limit the output to the top 6 key-value pairs
+    segmented_regions = segmented_regions[:6]
+    
     # Return the segmented images in descending order of similarity
     return segmented_regions
 
@@ -98,4 +101,4 @@ text_input = gr.Textbox(label="Text Input")  # Use Textbox with a label
 output_images = gr.outputs.JSON()
 
 # Create a Gradio interface
-gr.Interface(fn=segment_image, inputs=[input_image, text_input], outputs=output_images).launch(max_responses=6)
+gr.Interface(fn=segment_image, inputs=[input_image, text_input], outputs=output_images).launch()