Update app.py

app.py

@@ -8,7 +8,7 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from modules.classifyTags import classify_tags,process_tags
 from modules.reorganizer_model import reorganizer_list,reorganizer_class
 from modules.tag_enhancer import prompt_enhancer
-# from modules.florence2 import process_image,single_task_list,update_task_dropdown - Temporary disabled
+from modules.florence2 import process_image,single_task_list,update_task_dropdown
 
 os.environ['PYTORCH_ENABLE_MPS_FALLBACK']='1'
 
@@ -551,30 +551,30 @@ with gr.Blocks(title=TITLE, css=css, theme=gr.themes.Soft(), fill_width=True) as
                 character_mcut_enabled,
             ],
         )
-    # with gr.Tab(label="Florence 2 Image Captioning"):
-        # with gr.Row():
-            # with gr.Column(variant="panel"):
-                # input_img = gr.Image(label="Input Picture")
-                # task_type = gr.Radio(choices=['Single task', 'Cascaded task'], label='Task type selector', value='Single task')
-                # task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
-                # task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)
-                # text_input = gr.Textbox(label="Text Input (optional)")
-                # submit_btn = gr.Button(value="Submit")
-            # with gr.Column(variant="panel"):
-                # output_text = gr.Textbox(label="Output Text", show_label=True, show_copy_button=True, lines=8)
-                # output_img = gr.Image(label="Output Image")
-        # gr.Examples(
-            # examples=[
-                # ["images/image1.png", 'Object Detection'],
-                # ["images/image2.png", 'OCR with Region']
-            # ],
-            # inputs=[input_img, task_prompt],
-            # outputs=[output_text, output_img],
-            # fn=process_image,
-            # cache_examples=False,
-            # label='Try examples'
-        # )
-        # submit_btn.click(process_image, [input_img, task_prompt, text_input], [output_text, output_img])
+    with gr.Tab(label="Florence 2 Image Captioning"):
+        with gr.Row():
+            with gr.Column(variant="panel"):
+                input_img = gr.Image(label="Input Picture")
+                task_type = gr.Radio(choices=['Single task', 'Cascaded task'], label='Task type selector', value='Single task')
+                task_prompt = gr.Dropdown(choices=single_task_list, label="Task Prompt", value="Caption")
+                task_type.change(fn=update_task_dropdown, inputs=task_type, outputs=task_prompt)
+                text_input = gr.Textbox(label="Text Input (optional)")
+                submit_btn = gr.Button(value="Submit")
+            with gr.Column(variant="panel"):
+                output_text = gr.Textbox(label="Output Text", show_label=True, show_copy_button=True, lines=8)
+                output_img = gr.Image(label="Output Image")
+        gr.Examples(
+            examples=[
+                ["images/image1.png", 'Object Detection'],
+                ["images/image2.png", 'OCR with Region']
+            ],
+            inputs=[input_img, task_prompt],
+            outputs=[output_text, output_img],
+            fn=process_image,
+            cache_examples=False,
+            label='Try examples'
+        )
+        submit_btn.click(process_image, [input_img, task_prompt, text_input], [output_text, output_img])
     with gr.Tab("Booru Image Fetcher"):
         with gr.Row():
             with gr.Column():