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")
|