ISRI/ai-server
10
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: uploaded file were closed automatically before the model could infer their data

Browse source
This commit is contained in:
faraphel 2025-01-10 11:05:41 +01:00
parent 639425ad7d
commit b89fafdc96
6 changed files with 62 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
samples/models/dummy/config.json
View file

@ -2,5 +2,7 @@
"type": "python",
"file": "model.py",
"inputs": {}
"inputs": {
"file": {"type": "file"}
}
}

2
samples/models/dummy/model.py
View file

@ -8,5 +8,5 @@ def load(model) -> None:
def unload(model) -> None:
pass
def infer(model) -> typing.Iterator[bytes]:
def infer(model, file) -> typing.Iterator[bytes]:
yield json.dumps({"hello": "world!"}).encode("utf-8")