Update app.py

app.py

@@ -142,7 +142,7 @@ totensor = torchvision.transforms.ToTensor()
 normalize_fn = torchvision.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
 topilimage = torchvision.transforms.ToPILImage()
 
-def predict(input_1):
+def predict(input_1, input_2):
     im1 = normalize_fn(totensor(input_1))
     print(im1.shape)
     preds1 = model(im1.unsqueeze(0))/2 + 0.5