Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,8 +35,9 @@ def health():
|
|
| 35 |
return {"ok": True, "model": MODEL_ID}
|
| 36 |
|
| 37 |
@app.post("/generate")
|
| 38 |
-
def generate(file: UploadFile = File(...)):
|
| 39 |
-
|
|
|
|
| 40 |
inputs = image_processor(img, return_tensors="pt")
|
| 41 |
|
| 42 |
# Match encoder dtype/device (important when quantized/offloaded)
|
|
|
|
| 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)
|