Update app.py

app.py

@@ -4,9 +4,6 @@ from fastai.vision.learner import create_body
 import streamlit as st
 from PIL import Image
 import cv2 as cv
-
-# ---------Backend--------------------------------------------------------------
-
 import os
 import glob
 import time
@@ -35,7 +32,7 @@ class ColorizationDataset(Dataset):
         if split == 'train':
             self.transforms = transforms.Compose([
                 transforms.Resize((SIZE, SIZE),  Image.BICUBIC),
-                transforms.RandomHorizontalFlip(),  # A little data augmentation!
+                transforms.RandomHorizontalFlip(),
             ])
         elif split == 'val':
             self.transforms = transforms.Resize((SIZE, SIZE),  Image.BICUBIC)
@@ -59,7 +56,6 @@ class ColorizationDataset(Dataset):
         return len(self.paths)
 
 
-# A handy function to make our dataloaders
 def make_dataloaders(batch_size=16, n_workers=4, pin_memory=True, **kwargs):
     dataset = ColorizationDataset(**kwargs)
     dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=n_workers,
@@ -388,8 +384,6 @@ class MyDataset(torch.utils.data.Dataset):
         ab = img_lab[[1, 2], ...] / 110.
         return {'L': L, 'ab': ab}
 
-
-# A handy function to make our dataloaders
 def make_dataloaders2(batch_size=16, n_workers=4, pin_memory=True, **kwargs):
     dataset = MyDataset(**kwargs)
     dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=n_workers,