zdou0830 commited on
Commit
6ff3002
1 Parent(s): 971be17
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ fiber_demo = GLIPDemo(
51
 
52
  def predict(image, text, ground_tokens=""):
53
  ground_tokens = None if ground_tokens.strip() == "" else ground_tokens.strip().split(";")
54
- result, _ = glip_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, specified_tokens)
55
- fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, specified_tokens)
56
  return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]]
57
 
58
 
 
51
 
52
  def predict(image, text, ground_tokens=""):
53
  ground_tokens = None if ground_tokens.strip() == "" else ground_tokens.strip().split(";")
54
+ result, _ = glip_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens)
55
+ fiber_result, _ = fiber_demo.run_on_web_image(deepcopy(image[:, :, [2, 1, 0]]), text, 0.5, ground_tokens)
56
  return result[:, :, [2, 1, 0]], fiber_result[:, :, [2, 1, 0]]
57
 
58