Update app.py

app.py

@@ -46,9 +46,8 @@ class_names = [
 
 model.load_state_dict(torch.load('best_model.pth', map_location='cpu'))
 # Define a function to preprocess the input image
-def preprocess_input(input: str):
-    image_data = base64.b64decode(input)
-    image = Image.open(io.BytesIO(image_data))
+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))
@@ -58,7 +57,7 @@ def preprocess_input(input: str):
 
 # Define an endpoint to make predictions
 @app.post("/predict")
-async def predict_endpoint(input:str):
+async def predict_endpoint(input:fastapi.UploadFile):
     """Make a prediction on an image uploaded by the user."""
 
     # Preprocess the input image