csmithxc commited on
Commit
bfd9cc7
1 Parent(s): ab82199

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ from tools.demo import demo
21
  def parse_args():
22
  parser = argparse.ArgumentParser(
23
  description='YOLO-World Demo')
24
- parser.add_argument('--config', default='configs/pretrain/yolo_world_l_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
25
- parser.add_argument('--checkpoint', default='yolow-v8_l_clipv2_frozen_t2iv2_bn_o365_goldg_pretrain.pth')
26
  parser.add_argument(
27
  '--work-dir',
28
  help='the directory to save the file containing evaluation metrics')
 
21
  def parse_args():
22
  parser = argparse.ArgumentParser(
23
  description='YOLO-World Demo')
24
+ parser.add_argument('--config', default='configs/pretrain/yolo_world_s_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
25
+ parser.add_argument('--checkpoint', default='yolo_world_s_clip_base_dual_vlpan_2e-3adamw_32xb16_100e_o365_goldg_train_pretrained-18bea4d2.pth')
26
  parser.add_argument(
27
  '--work-dir',
28
  help='the directory to save the file containing evaluation metrics')