Update app.py

app.py

@@ -9,6 +9,7 @@ import torchvision.transforms as transforms
 from torch.utils.data import Dataset, DataLoader
 from PIL import Image
 import os
+import numpy as np
 
 # Set device to 'cpu' or 'cuda' if available
 device = torch.device('cpu')