jiuhai commited on
Commit
cc838f9
·
1 Parent(s): 549e76e
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -33,7 +33,7 @@ disable_btn = gr.Button(interactive=False)
33
  argparser = argparse.ArgumentParser()
34
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
35
  argparser.add_argument("--port", default="6324", type=str)
36
- argparser.add_argument("--model-path", default="umd-vt-nyu/clip-evaclip-und-gen-pretrain", type=str)
37
  argparser.add_argument("--model-base", type=str, default=None)
38
  argparser.add_argument("--num-gpus", type=int, default=1)
39
  argparser.add_argument("--conv-mode", type=str, default="plain")
@@ -121,7 +121,6 @@ def generate(state, imagebox, textbox, image_process_mode, temperature, top_p, m
121
 
122
  #images = [load_image_from_base64(image) for image in images]
123
  image_sizes = [image.size for image in images]
124
- print(images, image_processor, model.config)
125
 
126
  images = process_images(images, image_processor, model.config)
127
 
 
33
  argparser = argparse.ArgumentParser()
34
  argparser.add_argument("--server_name", default="0.0.0.0", type=str)
35
  argparser.add_argument("--port", default="6324", type=str)
36
+ argparser.add_argument("--model-path", default="umd-vt-nyu/clip-evaclip-und-gen-sft-3v", type=str)
37
  argparser.add_argument("--model-base", type=str, default=None)
38
  argparser.add_argument("--num-gpus", type=int, default=1)
39
  argparser.add_argument("--conv-mode", type=str, default="plain")
 
121
 
122
  #images = [load_image_from_base64(image) for image in images]
123
  image_sizes = [image.size for image in images]
 
124
 
125
  images = process_images(images, image_processor, model.config)
126