浏览代码

add `GET /models` endpoint

Jeffrey Morgan 1 年之前
父节点
当前提交
51164f57a2
共有 1 个文件被更改,包括 6 次插入0 次删除
  1. 6 0
      proto.py

+ 6 - 0
proto.py

@@ -74,6 +74,12 @@ def generate():
         stream_with_context(stream_response()), mimetype="text/event-stream"
         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__":
 if __name__ == "__main__":
     app.run(debug=True, threaded=True, port=5001)
     app.run(debug=True, threaded=True, port=5001)
     app.run()
     app.run()