Update app.py

app.py

@@ -5,6 +5,7 @@ from huggingface_hub import hf_hub_download
 import importlib.util
 from torchvision import transforms
 import random
+import numpy as np
 
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")