Upload 4 files

app.py

@@ -1,6 +1,20 @@
 import gradio as gr
 from feifeilib.feifeiflorencebase import process_image
 
+DESCRIPTION = "# [Florence-2 Image to Flux Prompt](https://huggingface.co/microsoft/Florence-2-base)"
+
+single_task_list =[
+    'Caption', 'Detailed Caption', 'More Detailed Caption', 'Object Detection',
+    'Dense Region Caption', 'Region Proposal', 'Caption to Phrase Grounding',
+    'Referring Expression Segmentation', 'Region to Segmentation',
+    'Open Vocabulary Detection', 'Region to Category', 'Region to Description',
+    'OCR', 'OCR with Region'
+]
+
+cascased_task_list =[
+    'Caption + Grounding', 'Detailed Caption + Grounding', 'More Detailed Caption + Grounding'
+]
+
 with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Tab(label="Florence-2 Image to Flux Prompt"):

feifeilib/feifeiflorencebase.py

@@ -25,8 +25,6 @@ processors = {
 }
 
 
-DESCRIPTION = "# [Florence-2 Image to Flux Prompt](https://huggingface.co/microsoft/Florence-2-base)"
-
 colormap = ['blue','orange','green','purple','brown','pink','gray','olive','cyan','red',
             'lime','indigo','violet','aqua','magenta','coral','gold','tan','skyblue']
 
@@ -215,20 +213,6 @@ def process_image(image, task_prompt = "More Detailed Caption", text_input=None,
     else:
         return "", None  # Return empty string and None for unknown task prompts
 
-
-single_task_list =[
-    'Caption', 'Detailed Caption', 'More Detailed Caption', 'Object Detection',
-    'Dense Region Caption', 'Region Proposal', 'Caption to Phrase Grounding',
-    'Referring Expression Segmentation', 'Region to Segmentation',
-    'Open Vocabulary Detection', 'Region to Category', 'Region to Description',
-    'OCR', 'OCR with Region'
-]
-
-cascased_task_list =[
-    'Caption + Grounding', 'Detailed Caption + Grounding', 'More Detailed Caption + Grounding'
-]
-
-
 def update_task_dropdown(choice):
     if choice == 'Cascased task':
         return gr.Dropdown(choices=cascased_task_list, value='Caption + Grounding')