Update utils/basic_functions.py

utils/basic_functions.py

@@ -130,8 +130,10 @@ def huggingface_detect_id_box(model_name,url):
         response_string = processor.decode(output[0][2:],skip_special_tokens=True)
         print(f"response_string: {response_string}")
 
-        bbox = re.search(r"ASSISTANT: \[(.*?)\]",response_string)
-        bbox = [image.size[0],image.size[1],image.size[0],image.size[1]]*ast.literal_eval([bbox])
+        match = re.search(r"ASSISTANT: \[(.*?)\]",response_string)
+        if not match:
+            return [image,"no match found"]
+        bbox = [image.size[0],image.size[1],image.size[0],image.size[1]]*ast.literal_eval([match.group(1)])
         print(f"bbox: {bbox}")
 
         
@@ -145,4 +147,4 @@ def huggingface_detect_id_box(model_name,url):
         return [image,bbox]
     except Exception as e:
         print(f"Error loading model or processing image: {str(e)}")
-        return None
+        return [image,"an error occurred processing request"]