Update utils/basic_functions.py

utils/basic_functions.py

@@ -127,7 +127,11 @@ def huggingface_detect_id_box(model_name,url):
         output = model.generate(**inputs,max_new_tokens=200,do_sample=False)
         print(f"output: {output}")
         
-        bbox = processor.decode(output[0][2:],skip_special_tokens=True)
+        response_string = processor.decode(output[0][2:],skip_special_tokens=True)
+        print(f"response_string: {response_string}")
+
+        bbox = re.search(r"ASSISTANT: \[(.*?)\]",text)
+        bbox = [image.size[0],image.size[1],image.size[0],image.size[1]]*ast.literal_eval([bbox])
         print(f"bbox: {bbox}")