diff --git a/src/routes/router.py b/src/routes/router.py index 00fdca4a33ffd480935b7556b42f4aebe54b7cde..9293e2da81b71c3adca2ed4c878dc54359ffbdc9 100644 --- a/src/routes/router.py +++ b/src/routes/router.py @@ -31,7 +31,9 @@ def setup_routes(app): @app.route('/render', methods=['GET']) def get_render(): - return render_controller.get_render() + hf_token = request.args.get('hf_token', '', type=str) + + return render_controller.get_render(hf_token) @app.route('/suggest', methods=['GET']) def get_suggest():