diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 205d5e31..afa37301 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -21,4 +21,6 @@ dependencies: - flask-ngrok - lupa==1.10 - transformers>=4.20.1 - - accelerate \ No newline at end of file + - accelerate + - flask-session + - python-socketio[client] \ No newline at end of file