Skip to content

Commit

Permalink
Do not change capitilization when saving
Browse files Browse the repository at this point in the history
  • Loading branch information
derneuere committed Dec 30, 2023
1 parent f34876d commit 59d8c2f
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions api/models/photo.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,9 +176,7 @@ def _generate_captions_im2txt(self, commit=True):
blip = True

caption = generate_caption(image_path=image_path, blip=blip, onnx=onnx)
caption = (
caption.replace("<start>", "").replace("<end>", "").strip().lower()
)
caption = caption.replace("<start>", "").replace("<end>", "").strip()
captions["im2txt"] = caption
self.captions_json = captions
# todo: handle duplicate captions
Expand Down

0 comments on commit 59d8c2f

Please # to comment.