Update app.py

app.py

@@ -47,7 +47,7 @@ ci = Interrogator(config)
 
 def inference(input_images, mode, best_max_flavors):
     # Process each image in the list and generate prompt results
-    prompt_results = []
+    prompt_texts = []
     for image_bytes in input_images:
         image = Image.open(io.BytesIO(image_bytes)).convert('RGB')
         if mode == 'best':
@@ -56,8 +56,9 @@ def inference(input_images, mode, best_max_flavors):
             prompt_result = ci.interrogate_classic(image)
         else:
             prompt_result = ci.interrogate_fast(image)
-        prompt_results.append((image, prompt_result))  # Use dictionary to set image labels
-    return prompt_results
+        prompt_texts.append(prompt_result['prompt'])  # Extract the prompt text
+    return prompt_texts
+
 
 
 title = """
@@ -150,7 +151,6 @@ a {text-decoration-line: underline; font-weight: 600;}
 }
 
 '''
-
 with gr.Blocks(css=css) as block:
     with gr.Column(elem_id="col-container"):
         gr.HTML(title)
@@ -162,9 +162,10 @@ with gr.Blocks(css=css) as block:
         
         submit_btn = gr.Button("Submit")
         
-        # rows, cols = NUM_IMAGES //3, 
+        # Display the prompt texts
         gallery = gr.Gallery(
-            label="Outputs", show_label=True, elem_id="gallery", object_fit="contain", height="auto"
+            label="Outputs", show_label=True, elem_id="gallery", object_fit="contain", height="auto",
+            src = "data:application/json,{{outputs['prompt_texts']}}"
         )
 
         with gr.Group(elem_id="share-btn-container"):
@@ -173,5 +174,5 @@ with gr.Blocks(css=css) as block:
         gr.HTML(article)
     submit_btn.click(fn=inference, inputs=[input_image,mode_input,flavor_input], outputs=[gallery], api_name="clipi2")
 
-    
-block.queue(max_size=32,concurrency_count=10).launch(show_api=False)
+block.queue(max_size=32,concurrency_count=10).launch(show_api=False)
+