weijiawu's picture
Duplicate from TencentARC/Caption-Anything
0ab9a32
raw
history blame contribute delete
179 Bytes
from text_refiner.text_refiner import TextRefiner
def build_text_refiner(type, device, args=None, api_key=""):
if type == 'base':
return TextRefiner(device, api_key)