Update app.py

app.py

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