Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
-import fastapi
-import numpy as np
+from fastapi import FastAPI, UploadFile
 from PIL import Image
+import numpy as np
 
 class TorchTensor(torch.Tensor):
     pass
@@ -12,8 +12,9 @@ class Prediction:
 app = fastapi.FastAPI(docs_url="/")
 
 model = torch.load("best_model.pth", map_location='cpu')
-# Define a function to preprocess the input image
-def preprocess_input(input: fastapi.UploadFile):
+
+def preprocess_input(input: UploadFile):
+    """Preprocess the input image."""
     image = Image.open(input.file)
     image = image.resize((224, 224))
     input = np.array(image)