test input

app.py

@@ -105,14 +105,15 @@ def deepseek(image, text_input, model_id):
     print(f"{prepare_inputs['sft_format'][0]}", answer)
     det_pattern = r"<\|det\|>\[\[(.+)]]<\|\/det\|>"
 
-    det_content = re.search(det_pattern, answer).group(1)
-    if det_content is None:
+    det_match = re.search(det_pattern, answer)
+    if det_match is None:
         return text_input, [], image
     
+    det_content = det_match.group(1)
     bbox = [ int(v.strip()) for v in det_content.split(",")]
     #w, h = image.size
     #bbox = [[bbox[0] * w, bbox[1] * h, bbox[2] * w, bbox[3] * h]]
-    
+
     scaled_boxes = rescale_bounding_boxes([bbox], image.width, image.height)
     return text_input, scaled_boxes, draw_bounding_boxes(image, scaled_boxes)