add GET /models
endpoint
This commit is contained in:
parent
6ba93652c6
commit
51164f57a2
1 changed files with 6 additions and 0 deletions
6
proto.py
6
proto.py
|
@ -74,6 +74,12 @@ def generate():
|
|||
stream_with_context(stream_response()), mimetype="text/event-stream"
|
||||
)
|
||||
|
||||
@app.route("/models", methods=["GET"])
|
||||
def models():
|
||||
all_files = os.listdir("./models")
|
||||
bin_files = [file.replace(".bin", "") for file in all_files if file.endswith(".bin")]
|
||||
return Response(json.dumps(bin_files), mimetype="application/json")
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(debug=True, threaded=True, port=5001)
|
||||
app.run()
|
||||
|
|
Loading…
Reference in a new issue