Spaces:

bcvilnrotter
/

id-code-identifier

Sleeping

bcvilnrotter commited on Feb 27, 2025

Commit

ca2edac

verified ·

1 Parent(s): 7cd0cdb

Update utils/basic_functions.py

Files changed (1) hide show

utils/basic_functions.py CHANGED Viewed

@@ -116,7 +116,8 @@ def huggingface_detect_id_box(model_name,url):
         """
         output = model.generate(**inputs,max_new_tokens=200,do_sample=False)
-        print(processor.decode(output[0][2:],skip_special_tokens=True))
         draw = ImageDraw.Draw(image)

         """
         output = model.generate(**inputs,max_new_tokens=200,do_sample=False)
+        bbox = processor.decode(output[0][2:],skip_special_tokens=True))
+        print(bbox)
         draw = ImageDraw.Draw(image)