markury commited on
Commit
41ffbc8
·
verified ·
1 Parent(s): 95bf31a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -287,11 +287,10 @@ def main():
287
  parser = argparse.ArgumentParser(description="Image Captioning Script")
288
  parser.add_argument('--input', '--folder_path', dest='folder_path', type=str, default='/content/images', help='Folder Path containing images')
289
  parser.add_argument('--caption_type', type=str, default='Descriptive', choices=list(CAPTION_TYPE_MAP.keys()), help='Caption Type')
290
- parser.add_argument('--length', '--caption_length', dest='caption_length', type=str, default='long', help='Caption Length')
291
  parser.add_argument('--extra_options', nargs='*', default=[], help='Extra Options')
292
  parser.add_argument('--name_input', type=str, default='', help='Person/Character Name (if applicable)')
293
  parser.add_argument('--custom_prompt', type=str, default='', help='Custom Prompt (optional)')
294
- parser.add_argument('--unwanted_start_texts_input', type=str, default='', help='Unwanted Starting Texts (comma-separated)')
295
 
296
  args = parser.parse_args()
297
 
 
287
  parser = argparse.ArgumentParser(description="Image Captioning Script")
288
  parser.add_argument('--input', '--folder_path', dest='folder_path', type=str, default='/content/images', help='Folder Path containing images')
289
  parser.add_argument('--caption_type', type=str, default='Descriptive', choices=list(CAPTION_TYPE_MAP.keys()), help='Caption Type')
290
+ parser.add_argument('--length', '--caption_length', dest='caption_length', type=str, default='short', help='Caption Length')
291
  parser.add_argument('--extra_options', nargs='*', default=[], help='Extra Options')
292
  parser.add_argument('--name_input', type=str, default='', help='Person/Character Name (if applicable)')
293
  parser.add_argument('--custom_prompt', type=str, default='', help='Custom Prompt (optional)')
 
294
 
295
  args = parser.parse_args()
296