Upload folder using huggingface_hub

Dockerfile

@@ -83,5 +83,7 @@ HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
 
 # Run the FastAPI server
 # The module path is constructed to work with the /app/env structure
+# --ws-ping-interval none --ws-ping-timeout none: Disable WebSocket ping/pong
+# to prevent timeout errors during long-running operations (e.g., token generation)
 ENV ENABLE_WEB_INTERFACE=true
-CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000"]
+CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000 --ws-ping-interval none --ws-ping-timeout none"]

server/app.py

@@ -79,11 +79,13 @@ def main(host: str = "0.0.0.0", port: int = 8000):
 
     For production deployments, consider using uvicorn directly with
     multiple workers:
-        uvicorn textarena_env.server.app:app --workers 4
+        uvicorn textarena_env.server.app:app --workers 4 --ws-ping-interval none --ws-ping-timeout none
     """
     import uvicorn
 
-    uvicorn.run(app, host=host, port=port)
+    # Disable WebSocket ping/pong to prevent timeout errors during
+    # long-running operations (e.g., token generation during training)
+    uvicorn.run(app, host=host, port=port, ws_ping_interval=None, ws_ping_timeout=None)
 
 
 if __name__ == "__main__":