Update app.py

app.py

@@ -62,12 +62,11 @@ torch.cuda.empty_cache()
 def process(input_image_editor, mask_image, input_text, strength, seed, randomize_seed, num_inference_steps, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
     if not input_text:
         raise gr.Error("Please enter a text prompt.")
-
+    item = gemini_predict(input_text)
+    
     image = input_image_editor['background']
-
     if not image:
         raise gr.Error("Please upload an image.")
-
     width, height = image.size
 
     if randomize_seed:
@@ -79,7 +78,8 @@ def process(input_image_editor, mask_image, input_text, strength, seed, randomiz
                           strength=strength, num_inference_steps=num_inference_steps, generator=generator,
                           guidance_scale=guidance_scale).images[0]
 
-    return result, mask_image, seed
+    
+    return result, mask_image, seed, item
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(MARKDOWN)
@@ -132,13 +132,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             submit_button_component = gr.Button(
                 value='Inpaint', variant='primary')
         with gr.Column(scale=1):
-            output_image_component = gr.Image(
-                type='pil', image_mode='RGB', label='Generated Image')
-            output_mask_component = gr.Image(
-                type='pil', image_mode='RGB', label='Generated Mask')
+            output_image_component = gr.Image(type='pil', image_mode='RGB', label='Generated Image')
+            output_mask_component = gr.Image(type='pil', image_mode='RGB', label='Generated Mask')
             with gr.Accordion("Debug Info", open=False):
                 output_seed = gr.Number(label="Used Seed")
-                gr.Textbox(gemini_predict(prompt))
+                identified_item = gr.Textbox(label="Gemini predicted item")
 
     submit_button_component.click(
         fn=process,