Update app.py

app.py

@@ -15,13 +15,13 @@ app = fastapi.FastAPI(docs_url="/")
 model = torch.load("model67.bin", map_location='cpu')
 
 # Define a function to preprocess the input image
-def preprocess_input(input: Any):
-    image = Image.open(BytesIO(input))
-    image = image.resize((224, 224))
-    input = np.array(image)
-    input = torch.from_numpy(input).float()
-    input = input.permute(2, 0, 1)
-    input = input.unsqueeze(0)
+def preprocess_input(input: fastapi.UploadFile):
+    image = Image.open(input.file)
+  image = image.resize((224, 224))
+  input = np.array(image)
+  input = torch.from_numpy(input).float()
+  input = input.permute(2, 0, 1)
+  input = input.unsqueeze(0)
     return input
 
 # Define an endpoint to make predictions