Add nn

app.py

@@ -4,6 +4,8 @@ import sys
 
 from IPython import display
 import torch
+from torch import nn
+from torch.nn import functional as F
 from torchvision import transforms
 from torchvision import utils as tv_utils
 from torchvision.transforms import functional as TF