lowercasing text

app.py

@@ -81,7 +81,7 @@ else:
     show = st.image(image)
     show.image(image, "Your Image")
     
-    gen_show_caption(sub, cap_prompt)
+    gen_show_caption(sub, imageLoader.text_transform(cap_prompt))
         
     image.close()
 

model.py

@@ -36,6 +36,9 @@ class ImageLoader():
     def transform(self, image):
         im = torch.FloatTensor(self.transformer(image)).unsqueeze(0)
         return {"image": im}
+    def text_transform(self, text):
+        # at present just lowercasing:
+        return text.lower()
     def to_image(self, tensor):
         return torchvision.transforms.ToPILImage()(tensor)