fix: bugs

Signed-off-by: ivelin <ivelin.eth@gmail.com>

app.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageDraw
 import math
 import torch
 import html
+import json
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 
 pretrained_repo_name = "ivelin/donut-refexp-draft"
@@ -55,6 +56,7 @@ def process_refexp(image: Image, prompt: str):
     print(
         fr"predicted decoder sequence before token2json: {html.escape(sequence)}")
     bbox = processor.token2json(sequence)
+    bbox = json.loads(bbox)
     print(f"predicted bounding box: {bbox}")
 
     print(f"image object: {image}")
@@ -63,10 +65,10 @@ def process_refexp(image: Image, prompt: str):
     print(f"image width, height: {width, height}")
     print(f"processed prompt: {prompt}")
 
-    xmin = math.floor(width*float(bbox["xmin"])) if bbox.get("xmin") else 0
-    ymin = math.floor(height*float(bbox["ymin"])) if bbox.get("ymin") else 0
-    xmax = math.floor(width*float(bbox["xmax"])) if bbox.get("xmax") else 1
-    ymax = math.floor(height*float(bbox["ymax"])) if bbox.get("ymax") else 1
+    xmin = math.floor(width*bbox["xmin"]) if bbox.get("xmin") else 0
+    ymin = math.floor(height*bbox["ymin"]) if bbox.get("ymin") else 0
+    xmax = math.floor(width*bbox["xmax"]) if bbox.get("xmax") else 1
+    ymax = math.floor(height*bbox["ymax"]) if bbox.get("ymax") else 1
 
     print(
         f"to image pixel values: xmin, ymin, xmax, ymax: {xmin, ymin, xmax, ymax}")