Update app.py

app.py

@@ -45,9 +45,9 @@ def process(
     image_input,
     image_prompter_input
 ) -> Tuple[Optional[Image.Image], Optional[str]]:
-    model = MODELS["microsoft/Florence-2-large-ft"]
-    processor = PROCESSORS["microsoft/Florence-2-large-ft"]
-    task = TASKS["microsoft/Florence-2-large-ft"]
+    model = MODELS[checkpoint_dropdown]
+    processor = PROCESSORS[checkpoint_dropdown]
+    task = TASKS[task_dropdown]
 
     if task_dropdown in IMAGE_TO_IMAGE_TASK_NAMES:
         _, response = run_inference(
@@ -89,7 +89,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         checkpoint_dropdown_component = gr.Dropdown(
             choices=CHECKPOINTS,
-            value=CHECKPOINTS[0],
+            value="microsoft/Florence-2-large-ft",
             label="Model", info="Select a Florence 2 model to use.",
             interactive=True
         )