Update app.py

app.py

@@ -28,6 +28,7 @@ def preprocess_input(input: fastapi.UploadFile):
     image = Image.open(input.file)
     image = image.resize((224, 224)).convert("RGB")
     input = np.array(image)
+    input = np.transpose(input, (2, 0, 1))
     input = torch.from_numpy(input).float()
     input = input.unsqueeze(0)
     return input