Docmatix-Florence-2

Running on Zero

andito HF Staff commited on Jun 21, 2024

Commit

b7139b3

1 Parent(s): 2b0ab69

test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def run_example(task_prompt, image, text_input=None):
         task=task_prompt,
         image_size=(image.width, image.height)
     )
-    return parsed_answer[task_prompt].replace("<pad>", "")
 def plot_bbox(image, data):
     fig, ax = plt.subplots()

         task=task_prompt,
         image_size=(image.width, image.height)
     )
+    return parsed_answer[task_prompt]#.replace("<pad>", "")
 def plot_bbox(image, data):
     fig, ax = plt.subplots()