fix bugs

app.py

@@ -48,7 +48,7 @@ def fig_to_pil(fig):
     return Image.open(buf)
 
 
-def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-large'):
+def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florence-2-base'):
     with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
         model = models[model_id]
     processor = processors[model_id]
@@ -127,7 +127,7 @@ def draw_ocr_bboxes(image, prediction):
                   fill=color)
     return image
 
-def process_image(image, task_prompt, text_input=None, model_id='microsoft/Florence-2-large'):
+def process_image(image, task_prompt, text_input=None, model_id='microsoft/Florence-2-base'):
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
     if task_prompt == 'Caption':
         task_prompt = '<CAPTION>'
@@ -263,7 +263,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(label="Input Picture")
-                model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='microsoft/Florence-2-large')
+                model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value='microsoft/Florence-2-base')
                 task_type = gr.Radio(choices=['Single task', 'Cascased 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)

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-timm
+timm
+einops