Update app.py

app.py

@@ -3,6 +3,7 @@ import torchvision.transforms as transforms
 from PIL import Image
 import torch
 from timm.models import create_model
+import numpy as np
 def predict(input_img):
     input_img =  Image.fromarray(np.uint8(input_img))
     model1 = create_model(