Update app.py

app.py

@@ -15,13 +15,8 @@ async def classify_image(file: UploadFile = File(...)):
         # Read the file contents into a PIL image
         image = Image.open(file.file).convert('RGB')
 
-        # Convert the image to bytes
-        image_bytes = io.BytesIO()
-        image.save(image_bytes, format='JPEG')
-        image_bytes = image_bytes.getvalue()
-
         # Perform image classification
-        result = pipe(image_bytes)
+        result = pipe(image)
 
         return {"classification_result": result[0]}  # Return the top prediction