Upload folder using huggingface_hub

src/app.py

@@ -10,7 +10,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from litrl.env.connect_four import Board
 from loguru import logger
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, RedirectResponse
 from src.app_state import AppState
 from src.typing import CpuConfig
 from src.huggingface.huggingface_client import HuggingFaceClient
@@ -26,6 +26,10 @@ def stream_mp4(mp4_path: Path) -> StreamingResponse:
 def create_app() -> FastAPI:
     app = FastAPI()
 
+    @app.get('/')
+    async def to_docs():
+        return RedirectResponse("/docs")
+
     @app.post("/", response_model=int)
     def bot_action(
         board: Board,
@@ -65,7 +69,7 @@ def create_app() -> FastAPI:
     @app.get(path=f"/mp4")
     def bot_action(
         env_id: SingleAgentId,
-    ) -> str:
+    ) -> StreamingResponse:
         env = make(id=env_id, render_mode="rgb_array")
         env = RecordVideo(
             env=env,