From 373f92ee1dd05e26bdb744bb634f430d4a09ca9a Mon Sep 17 00:00:00 2001
From: "sweep-nightly[bot]"
 <131841235+sweep-nightly[bot]@users.noreply.github.com>
Date: Mon, 23 Oct 2023 00:48:29 +0000
Subject: [PATCH] Sandbox run src/api.py

---
 src/api.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/api.py b/src/api.py
index db10957..8456fff 100644
--- a/src/api.py
+++ b/src/api.py
@@ -1,7 +1,8 @@
-from fastapi import FastAPI, UploadFile, File
-from PIL import Image
 import torch
+from fastapi import FastAPI, File, UploadFile
+from PIL import Image
 from torchvision import transforms
+
 from main import MNISTTrainer  # Importing MNISTTrainer class from main.py
 
 # Load the model
@@ -11,13 +12,13 @@
 model.eval()
 
 # Transform used for preprocessing the image
-transform = transforms.Compose([
-    transforms.ToTensor(),
-    transforms.Normalize((0.5,), (0.5,))
-])
+transform = transforms.Compose(
+    [transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))]
+)
 
 app = FastAPI()
 
+
 @app.post("/predict/")
 async def predict(file: UploadFile = File(...)):
     image = Image.open(file.file).convert("L")