justinj92 commited on
Commit
833e9bd
·
verified ·
1 Parent(s): b7b4f61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,9 +45,9 @@ def process(
45
  image_input,
46
  image_prompter_input
47
  ) -> Tuple[Optional[Image.Image], Optional[str]]:
48
- model = MODELS[checkpoint_dropdown]
49
- processor = PROCESSORS[checkpoint_dropdown]
50
- task = TASKS[task_dropdown]
51
 
52
  if task_dropdown in IMAGE_TO_IMAGE_TASK_NAMES:
53
  _, response = run_inference(
 
45
  image_input,
46
  image_prompter_input
47
  ) -> Tuple[Optional[Image.Image], Optional[str]]:
48
+ model = MODELS["microsoft/Florence-2-large-ft"]
49
+ processor = PROCESSORS["microsoft/Florence-2-large-ft"]
50
+ task = TASKS["microsoft/Florence-2-large-ft"]
51
 
52
  if task_dropdown in IMAGE_TO_IMAGE_TASK_NAMES:
53
  _, response = run_inference(