Update app.py

app.py

@@ -103,11 +103,13 @@ def segment_image(input_image, text_input):
         api_result = client.predict(image_base64, text_input, api_name="/predict")
 
         # Append the API result and similarity to the list
-        results_with_similarity.append({"api_result": api_result, "similarity": similarity})
+        results_with_similarity.append({"api_result": api_result, "image": image_base64, "similarity": similarity})
 
+    results_with_similarity = results_with_similarity[:1]
+        
     # Print the top 6 results
-    for result in results_with_similarity:
-        print(result)
+    # for result in results_with_similarity:
+    #     print(result)
     
     # Return the segmented images in descending order of similarity
     return results_with_similarity
@@ -115,7 +117,7 @@ def segment_image(input_image, text_input):
 # Create Gradio components
 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()
+output_images = gr.outputs.JSON()
 
 # Create a Gradio interface
-gr.Interface(fn=segment_image, inputs=[input_image, text_input], outputs="text").launch()
+gr.Interface(fn=segment_image, inputs=[input_image, text_input], outputs=output_images).launch()