Update feifeiui/feifeiui.py

feifeiui/feifeiui.py

@@ -6,6 +6,7 @@ from feifeilib.feifeichat import feifeichat
 from feifeilib.feifeitexttoimg import feifeitexttoimg
 from feifeilib.feifeiflorence import feifeiflorence
 from feifeilib.feifeifluxapi import feifeifluxapi
+from feifeilib.feifeiflorencebase import process_image
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -129,7 +130,15 @@ def create_ui():
                     output_text = gr.Textbox(label="Output Text",
                                              show_label=False,
                                              container=False)
-
+                    
+                with gr.Tab(label="Florence-2"):
+                    with gr.Row():
+                        florence_input_img = gr.Image(label="Input Picture",height=320)
+                    with gr.Row():
+                        florence_submit_btn = gr.Button(value="Submit")
+                    with gr.Row():
+                        florence_output_text = gr.Textbox(label="Output Text")
+                        
             with gr.Column(scale=1, elem_id="col-container"):
                 gr.ChatInterface(
                     feifeichat,
@@ -181,4 +190,6 @@ def create_ui():
             inputs=[input_img],  # Input components for the function
             outputs=[output_text],  # Output component for the function
         )
+
+        florence_submit_btn.click(process_image, [florence_input_img], [florence_output_text])
     return FeiFei