ttengwang commited on
Commit
3421695
1 Parent(s): 0f9d3ab

fix text_refiner=None

Browse files
Files changed (1) hide show
  1. 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 and text_refiner is not None:
20
- self.init_refiner(api_key)
 
 
 
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: