Upload app.py

app.py

@@ -4,7 +4,7 @@ from typing import Dict
 import numpy as np
 import tensorflow as tf
 from PIL import Image
-from bustapi import BustAPI, Request, Response
+from bustapi import BustAPI, request, Response
 
 LABELS = [
     "battery",        # 电池
@@ -93,9 +93,9 @@ def predict(image):
 app = BustAPI()
 
 @app.post("/predict")
-async def predict_api(req: Request):
+async def predict_api():
     # 读取 POST 二进制流
-    img_bytes = await req.body()
+    img_bytes = await request.body()
     # 转 PIL Image
     image = Image.open(io.BytesIO(img_bytes)).convert("RGB")
     # 推理