updated laoder

model.py

@@ -26,6 +26,13 @@ class ImageLoader():
                                torchvision.transforms.Resize(256),
                                torchvision.transforms.CenterCrop(224),
                                torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225))])
+        self.show_manip = torchvision.transforms.Compose(
+            [
+                torchvision.transforms.ToTensor(),
+                torchvision.transforms.Resize(500, max_size=500),
+                torchvision.transforms.ToPILImage()
+            ]
+        )
         
     def load(self, im_path):
         im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)
@@ -44,12 +51,8 @@ class ImageLoader():
         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)
+        return self.show_manip(image)
     
-    def to_image(self, tensor):
-        return torchvision.transforms.ToPILImage()(tensor)
 
 class VirTexModel():
     def __init__(self):