Spaces:

zdou0830
/

desco

Sleeping

zdou0830 commited on Dec 4, 2023

Commit

bbe0905

•

1 Parent(s): 2a55a2e

specified tokens

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,7 @@ def predict(image, text, specified_tokens=""):
     if specified_tokens.strip() == "":
         specified_tokens = None
     else:
-        specified_tokens = specified_tokens..split(";")
     result, _ = glip_demo.run_on_web_image(image[:, :, [2, 1, 0]], text, 0.5, specified_tokens)
     return result[:, :, [2, 1, 0]]

     if specified_tokens.strip() == "":
         specified_tokens = None
     else:
+        specified_tokens = specified_tokens.strip().split(";")
     result, _ = glip_demo.run_on_web_image(image[:, :, [2, 1, 0]], text, 0.5, specified_tokens)
     return result[:, :, [2, 1, 0]]