Update app.py
Browse files
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='
|
| 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 |
|