Parsa2025AI commited on
Commit
ada6a86
·
verified ·
1 Parent(s): 9a01c15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -35,9 +35,8 @@ def health():
35
  return {"ok": True, "model": MODEL_ID}
36
 
37
  @app.post("/generate")
38
- async def generate(file: UploadFile = File(...)):
39
- data = await file.read()
40
- img = Image.open(io.BytesIO(data)).convert("RGB")
41
  inputs = image_processor(img, return_tensors="pt")
42
 
43
  # Match encoder dtype/device (important when quantized/offloaded)
 
35
  return {"ok": True, "model": MODEL_ID}
36
 
37
  @app.post("/generate")
38
+ def generate(file: UploadFile = File(...)):
39
+ img = Image.open(io.BytesIO(file.file.read())).convert("RGB")
 
40
  inputs = image_processor(img, return_tensors="pt")
41
 
42
  # Match encoder dtype/device (important when quantized/offloaded)