[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Этот коммит содержится в:
pre-commit-ci[bot] 2023-02-23 14:02:30 +00:00
родитель c136b91fba
Коммит 28014f563f

Просмотреть файл

@ -58,9 +58,7 @@ class SummaryDetector(AnalysisMethod):
self.image_summary["const_image_summary"] = summary_model.generate( self.image_summary["const_image_summary"] = summary_model.generate(
{"image": image} {"image": image}
)[0] )[0]
self.image_summary[ self.image_summary["3_non-deterministic summary"] = summary_model.generate(
"3_non-deterministic summary"
] = summary_model.generate(
{"image": image}, use_nucleus_sampling=True, num_captions=3 {"image": image}, use_nucleus_sampling=True, num_captions=3
) )
for key in self.image_summary: for key in self.image_summary: