Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,9 +35,8 @@ def health():
|
|
| 35 |
return {"ok": True, "model": MODEL_ID}
|
| 36 |
|
| 37 |
@app.post("/generate")
|
| 38 |
-
|
| 39 |
-
|
| 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)
|