niulx commited on
Commit
dce7d46
1 Parent(s): a3221b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -266,7 +266,7 @@ with gr.Blocks() as demo:
266
  run_main,
267
  mask_np_list=mask_np_list,
268
  mask_label_list=mask_label_list,
269
- image_gt=image,
270
  num_tokens=int(num_tokens),
271
  embedding_learning_rate = float(embedding_learning_rate),
272
  max_emb_train_steps = int(max_emb_train_steps),
 
266
  run_main,
267
  mask_np_list=mask_np_list,
268
  mask_label_list=mask_label_list,
269
+ image_gt=np.array(image),
270
  num_tokens=int(num_tokens),
271
  embedding_learning_rate = float(embedding_learning_rate),
272
  max_emb_train_steps = int(max_emb_train_steps),