resizing to 500 for image display

app.py

@@ -77,9 +77,10 @@ else:
     image_dict = imageLoader.transform(image)
     
 #     image = imageLoader.to_image(image_dict["image"].squeeze(0))
+    show_image = imageLoader.show_resize(image)
     
-    show = st.image(image)
-    show.image(image, "Your Image")
+    show = st.image(show_image)
+    show.image(show_image, "Your Image")
     
     gen_show_caption(sub, imageLoader.text_transform(cap_prompt))
         

model.py

@@ -30,15 +30,24 @@ class ImageLoader():
     def load(self, im_path):
         im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)
         return {"image": im}
+    
     def raw_load(self, im_path):
         im = torch.FloatTensor(Image.open(im_path))
         return {"image": im}
+    
     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 show_resize(self, image):
+        im = torchvision.transforms.functional.pil_to_transform(image)
+        im = torchvision.transforms.functional.resize(image, size=500, max_size=500)
+        return self.to_image(im)
+    
     def to_image(self, tensor):
         return torchvision.transforms.ToPILImage()(tensor)