Update app.py

app.py

@@ -1,5 +1,5 @@
 import fastapi
-from fastapi import FastAPI
+from fastapi import FastAPI,UploadFile
 from transformers import AutoModelForImageClassification
 app = FastAPI()
 # Load the model from the local file
@@ -21,7 +21,7 @@ def preprocess_image(image):
 
 # Define an endpoint to predict the output
 @app.post("/predict")
-async def predict_endpoint(image: fastapi.File):
+async def predict_endpoint(image: UploadFile):
   # Preprocess the image
   image = preprocess_image(image)