Spaces:

stepfun-ai
/

GOT_official_online_demo

Running on Zero

ucaslcl commited on Sep 14

Commit

5a724fe

•

1 Parent(s): 07b5d13

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,6 +11,7 @@ model = model.eval().cuda()
 @spaces.GPU
 def run_GOT(image_array, got_mode, ocr_box="", ocr_color=""):
     print("image_array: ", image_array)
     # image = Image.fromarray(np.uint8(image_array))
     image = image_array
     if got_mode == "plain texts OCR":

 @spaces.GPU
 def run_GOT(image_array, got_mode, ocr_box="", ocr_color=""):
     print("image_array: ", image_array)
+    print(got_mode, ' ', ocr_box, ' ', ocr_color)
     # image = Image.fromarray(np.uint8(image_array))
     image = image_array
     if got_mode == "plain texts OCR":