Update app/main.py

app/main.py

@@ -6,6 +6,7 @@ from .food_model import classify_food
 from .fruit_model import classify_fruit
 
 app = FastAPI()
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -27,4 +28,4 @@ async def predict_food(file: UploadFile = File(...)):
 async def predict_fruit(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(await file.read())).convert("RGB")
     result = classify_fruit(image)
-    return {"prediction": result}
+    return {"prediction": result}