added to tesnor

model.py

@@ -22,6 +22,7 @@ SAMPLES_PATH = "./samples/*.jpg"
 class ImageLoader():
     def __init__(self):
         self.transformer = torchvision.transforms.Compose([
+                               torchvision.transforms.ToTensor(),
                                torchvision.transforms.Resize(256),
                                torchvision.transforms.CenterCrop(224),
                                torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225)),
@@ -29,9 +30,8 @@ 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)).unsqueeze(0)
+        im = torch.FloatTensor(Image.open(im_path))
         return {"image": im}
     def transform(self, image):
         im = torch.FloatTensor(self.transformer(image)).unsqueeze(0)