ソースを参照

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"
     )
 
+@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()