Update app.py

app.py

@@ -5,6 +5,8 @@ from torch.utils.data import DataLoader
 from torchvision import datasets
 from torchvision.transforms import ToTensor
 import torch.nn.functional as F
+import numpy as np
+from PIL import Image
 
 device = (
     "cuda"