fix: cleanup

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

app.py

@@ -46,6 +46,7 @@ def process_refexp(image: Image, prompt: str):
 
     # postprocess
     sequence = processor.batch_decode(outputs.sequences)[0]
+    print(f"predicted decoder sequence: {sequence}")
     sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(
         processor.tokenizer.pad_token, "")
     # remove first task start token
@@ -59,10 +60,10 @@ def process_refexp(image: Image, prompt: str):
     print(f"image width, height: {width, height}")
     print(f"processed prompt: {prompt}")
 
-    xmin = math.floor(width*bbox["xmin"])
-    ymin = math.floor(height*bbox["ymin"])
-    xmax = math.floor(width*bbox["xmax"])
-    ymax = math.floor(height*bbox["ymax"])
+    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}")