Update app.py

imported transfrom

app.py

@@ -7,6 +7,7 @@ import tensorflow as tf
 import math
 import torch.nn.functional as F
 from tqdm.auto import tqdm
+from torchvision import transforms
 
 from torch import nn
 img_size = 64