File size: 843 Bytes
acb3380 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from ctm.processors.processor_gpt4v import GPT4VProcessor
@GPT4VProcessor.register_processor("gpt4v_cloth_fashion_processor") # type: ignore[no-untyped-call] # FIX ME
class GPT4VClothFashionProcessor(GPT4VProcessor):
def __init__(self, *args, **kwargs): # type: ignore[no-untyped-def] # FIX ME
self.init_processor() # type: ignore[no-untyped-call] # FIX ME
self.task_instruction = "Focus on the cloth of people in the image, describe the style of the cloth fashion. If there is no people detected, please answer with None."
if __name__ == "__main__":
processor = GPT4VProcessor("close_fashion_processor") # type: ignore[no-untyped-call] # FIX ME
image_path = "../ctmai-test1.png"
summary: str = processor.ask_info(query=None, image_path=image_path) # type: ignore[arg-type] # FIX ME
print(summary)
|