Skip to content

Commit e3a9f13

Browse files
committed
addressed comments
1 parent 38ccc91 commit e3a9f13

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

clarifai_datautils/multimodal/pipeline/summarizer.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,16 +24,17 @@ class ImageSummarizer(BaseTransform):
2424
""" Summarizes image elements. """
2525

2626
def __init__(self,
27-
pat,
28-
model_url="https://clarifai.com/qwen/qwen-VL/models/qwen-VL-Chat",
29-
prompt=SUMMARY_PROMPT):
27+
model_url: str = "https://clarifai.com/qwen/qwen-VL/models/qwen-VL-Chat",
28+
pat: str = None,
29+
prompt: str = SUMMARY_PROMPT):
3030
"""Initializes an ImageSummarizer object.
3131
3232
Args:
3333
pat (str): Clarifai PAT.
3434
model_url (str): Model URL to use for summarization.
3535
prompt (str): Prompt to use for summarization.
3636
"""
37+
self.pat = pat
3738
self.model_url = model_url
3839
self.model = Model(url=model_url, pat=pat)
3940
self.summary_prompt = prompt

0 commit comments

Comments
 (0)