Spaces:
Build error
Build error
Update caption_anything.py
Browse files- caption_anything.py +1 -1
caption_anything.py
CHANGED
@@ -62,7 +62,7 @@ def parse_augment():
|
|
62 |
parser.add_argument('--clip_filter', action="store_true", help="use clip to filter bad captions")
|
63 |
parser.add_argument('--context_captions', action="store_true", help="use surrounding captions to enhance current caption (TODO)")
|
64 |
parser.add_argument('--disable_regular_box', action="store_true", default = False, help="crop image with a regular box")
|
65 |
-
parser.add_argument('--device', type=str, default="
|
66 |
parser.add_argument('--port', type=int, default=6086, help="only useful when running gradio applications")
|
67 |
parser.add_argument('--debug', action="store_true")
|
68 |
parser.add_argument('--gradio_share', action="store_true")
|
|
|
62 |
parser.add_argument('--clip_filter', action="store_true", help="use clip to filter bad captions")
|
63 |
parser.add_argument('--context_captions', action="store_true", help="use surrounding captions to enhance current caption (TODO)")
|
64 |
parser.add_argument('--disable_regular_box', action="store_true", default = False, help="crop image with a regular box")
|
65 |
+
parser.add_argument('--device', type=str, default="cpu")
|
66 |
parser.add_argument('--port', type=int, default=6086, help="only useful when running gradio applications")
|
67 |
parser.add_argument('--debug', action="store_true")
|
68 |
parser.add_argument('--gradio_share', action="store_true")
|