Spaces:
Runtime error
Runtime error
ttengwang
commited on
Commit
·
3421695
1
Parent(s):
0f9d3ab
fix text_refiner=None
Browse files- caption_anything.py +5 -2
caption_anything.py
CHANGED
@@ -16,8 +16,11 @@ class CaptionAnything():
|
|
16 |
self.segmenter = build_segmenter(args.segmenter, args.device, args) if segmenter is None else segmenter
|
17 |
|
18 |
self.text_refiner = None
|
19 |
-
if not args.disable_gpt
|
20 |
-
|
|
|
|
|
|
|
21 |
|
22 |
def init_refiner(self, api_key):
|
23 |
try:
|
|
|
16 |
self.segmenter = build_segmenter(args.segmenter, args.device, args) if segmenter is None else segmenter
|
17 |
|
18 |
self.text_refiner = None
|
19 |
+
if not args.disable_gpt:
|
20 |
+
if text_refiner is not None:
|
21 |
+
self.text_refiner = text_refiner
|
22 |
+
else:
|
23 |
+
self.init_refiner(api_key)
|
24 |
|
25 |
def init_refiner(self, api_key):
|
26 |
try:
|