Upgrade deps and migrate to Gradio 6

README.md

@@ -4,7 +4,7 @@ emoji: 👁
 colorFrom: blue
 colorTo: red
 sdk: gradio
-sdk_version: 5.44.1
+sdk_version: 6.13.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -30,7 +30,7 @@ def get_sample_image_markdown(name: str) -> str:
 
 model = Model()
 
-with gr.Blocks(css_paths="style.css") as demo:
+with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
 
     with gr.Tabs():
@@ -71,4 +71,4 @@ with gr.Blocks(css_paths="style.css") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css_paths="style.css")

pyproject.toml

@@ -5,11 +5,15 @@ description = ""
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "gradio>=5.44.1",
-    "hf-transfer>=0.1.9",
-    "timm>=1.0.19",
-    "torch==2.8.0",
-    "torchvision>=0.23.0",
+    "gradio[mcp,oauth]>=6.13.0",
+    "timm>=1.0.26",
+    "torch==2.9.1",
+    "torchvision>=0.24.1",
+]
+
+[dependency-groups]
+hf-spaces = [
+    "datasets",
 ]
 
 [tool.ruff]

requirements.txt

@@ -1,232 +1,365 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml -o requirements.txt
-aiofiles==24.1.0
-    # via gradio
+#    uv export --no-hashes --no-dev --group hf-spaces --no-emit-package typer-slim --no-emit-package spaces -o requirements.txt
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.13.5
+    # via fsspec
+aiosignal==1.4.0
+    # via aiohttp
+annotated-doc==0.0.4
+    # via
+    #   fastapi
+    #   typer
 annotated-types==0.7.0
     # via pydantic
-anyio==4.10.0
+anyio==4.13.0
     # via
     #   gradio
     #   httpx
+    #   mcp
+    #   sse-starlette
     #   starlette
-brotli==1.1.0
+async-timeout==5.0.1 ; python_full_version < '3.11'
+    # via aiohttp
+attrs==26.1.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+audioop-lts==0.2.2 ; python_full_version >= '3.13'
+    # via gradio
+authlib==1.7.0
     # via gradio
-certifi==2025.8.3
+brotli==1.2.0
+    # via gradio
+certifi==2026.4.22
     # via
     #   httpcore
     #   httpx
     #   requests
-charset-normalizer==3.4.3
+cffi==2.0.0 ; platform_python_implementation != 'PyPy'
+    # via cryptography
+charset-normalizer==3.4.7
     # via requests
-click==8.2.1
+click==8.3.3
     # via
     #   typer
     #   uvicorn
-exceptiongroup==1.3.0
+colorama==0.4.6 ; sys_platform == 'win32'
+    # via
+    #   click
+    #   tqdm
+cryptography==46.0.7
+    # via
+    #   authlib
+    #   joserfc
+    #   pyjwt
+datasets==4.8.4
+dill==0.4.1
+    # via
+    #   datasets
+    #   multiprocess
+exceptiongroup==1.3.1 ; python_full_version < '3.11'
     # via anyio
-fastapi==0.116.1
+fastapi==0.136.1
     # via gradio
-ffmpy==0.6.1
-    # via gradio
-filelock==3.19.1
+filelock==3.29.0
     # via
+    #   datasets
     #   huggingface-hub
     #   torch
-fsspec==2025.9.0
+frozenlist==1.8.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2026.2.0
     # via
+    #   datasets
     #   gradio-client
     #   huggingface-hub
     #   torch
-gradio==5.44.1
-    # via styleswin (pyproject.toml)
-gradio-client==1.12.1
-    # via gradio
+gradio==6.13.0
+    # via styleswin
+gradio-client==2.5.0
+    # via
+    #   gradio
+    #   hf-gradio
 groovy==0.1.2
     # via gradio
 h11==0.16.0
     # via
     #   httpcore
     #   uvicorn
-hf-transfer==0.1.9
-    # via styleswin (pyproject.toml)
-hf-xet==1.1.9
+hf-gradio==0.4.1
+    # via gradio
+hf-xet==1.4.3 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
     # via huggingface-hub
 httpcore==1.0.9
     # via httpx
 httpx==0.28.1
     # via
+    #   datasets
     #   gradio
     #   gradio-client
+    #   huggingface-hub
+    #   mcp
     #   safehttpx
-huggingface-hub==0.34.4
+httpx-sse==0.4.3
+    # via mcp
+huggingface-hub==1.11.0
     # via
+    #   datasets
     #   gradio
     #   gradio-client
     #   timm
-idna==3.10
+idna==3.13
     # via
     #   anyio
     #   httpx
     #   requests
+    #   yarl
+itsdangerous==2.2.0
+    # via gradio
 jinja2==3.1.6
     # via
     #   gradio
     #   torch
+joserfc==1.6.4
+    # via authlib
+jsonschema==4.26.0
+    # via mcp
+jsonschema-specifications==2025.9.1
+    # via jsonschema
 markdown-it-py==4.0.0
     # via rich
-markupsafe==3.0.2
+markupsafe==3.0.3
     # via
     #   gradio
     #   jinja2
+mcp==1.27.0
+    # via gradio
 mdurl==0.1.2
     # via markdown-it-py
 mpmath==1.3.0
     # via sympy
-networkx==3.4.2
+multidict==6.7.1
+    # via
+    #   aiohttp
+    #   yarl
+multiprocess==0.70.19
+    # via datasets
+networkx==3.4.2 ; python_full_version < '3.11'
+    # via torch
+networkx==3.6.1 ; python_full_version >= '3.11'
     # via torch
-numpy==2.2.6
+numpy==2.2.6 ; python_full_version < '3.11'
     # via
+    #   datasets
     #   gradio
     #   pandas
     #   torchvision
-nvidia-cublas-cu12==12.8.4.1
+numpy==2.4.4 ; python_full_version >= '3.11'
+    # via
+    #   datasets
+    #   gradio
+    #   pandas
+    #   torchvision
+nvidia-cublas-cu12==12.8.4.1 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via
     #   nvidia-cudnn-cu12
     #   nvidia-cusolver-cu12
     #   torch
-nvidia-cuda-cupti-cu12==12.8.90
+nvidia-cuda-cupti-cu12==12.8.90 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cuda-nvrtc-cu12==12.8.93
+nvidia-cuda-nvrtc-cu12==12.8.93 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cuda-runtime-cu12==12.8.90
+nvidia-cuda-runtime-cu12==12.8.90 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cudnn-cu12==9.10.2.21
+nvidia-cudnn-cu12==9.10.2.21 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cufft-cu12==11.3.3.83
+nvidia-cufft-cu12==11.3.3.83 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cufile-cu12==1.13.1.3
+nvidia-cufile-cu12==1.13.1.3 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-curand-cu12==10.3.9.90
+nvidia-curand-cu12==10.3.9.90 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cusolver-cu12==11.7.3.90
+nvidia-cusolver-cu12==11.7.3.90 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-cusparse-cu12==12.5.8.93
+nvidia-cusparse-cu12==12.5.8.93 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via
     #   nvidia-cusolver-cu12
     #   torch
-nvidia-cusparselt-cu12==0.7.1
+nvidia-cusparselt-cu12==0.7.1 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-nccl-cu12==2.27.3
+nvidia-nccl-cu12==2.27.5 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvjitlink-cu12==12.8.93 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via
     #   nvidia-cufft-cu12
     #   nvidia-cusolver-cu12
     #   nvidia-cusparse-cu12
     #   torch
-nvidia-nvtx-cu12==12.8.90
+nvidia-nvshmem-cu12==3.3.20 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-orjson==3.11.3
+nvidia-nvtx-cu12==12.8.90 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+    # via torch
+orjson==3.11.8
     # via gradio
-packaging==25.0
+packaging==26.1
     # via
+    #   datasets
     #   gradio
     #   gradio-client
     #   huggingface-hub
-pandas==2.3.2
-    # via gradio
-pillow==11.3.0
+pandas==2.3.3 ; python_full_version < '3.11'
+    # via
+    #   datasets
+    #   gradio
+pandas==3.0.2 ; python_full_version >= '3.11'
+    # via
+    #   datasets
+    #   gradio
+pillow==12.2.0
     # via
     #   gradio
     #   torchvision
-pydantic==2.11.7
+propcache==0.4.1
+    # via
+    #   aiohttp
+    #   yarl
+pyarrow==24.0.0
+    # via datasets
+pycparser==3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy'
+    # via cffi
+pydantic==2.12.5
     # via
     #   fastapi
     #   gradio
-pydantic-core==2.33.2
+    #   mcp
+    #   pydantic-settings
+pydantic-core==2.41.5
     # via pydantic
+pydantic-settings==2.14.0
+    # via mcp
 pydub==0.25.1
     # via gradio
-pygments==2.19.2
+pygments==2.20.0
     # via rich
+pyjwt==2.12.1
+    # via mcp
 python-dateutil==2.9.0.post0
     # via pandas
-python-multipart==0.0.20
-    # via gradio
-pytz==2025.2
-    # via pandas
-pyyaml==6.0.2
+python-dotenv==1.2.2
+    # via pydantic-settings
+python-multipart==0.0.26
     # via
     #   gradio
+    #   mcp
+pytz==2026.1.post1
+    # via
+    #   gradio
+    #   pandas
+pywin32==311 ; sys_platform == 'win32'
+    # via mcp
+pyyaml==6.0.3
+    # via
+    #   datasets
+    #   gradio
     #   huggingface-hub
     #   timm
-requests==2.32.5
-    # via huggingface-hub
-rich==14.1.0
+referencing==0.37.0
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+requests==2.33.1
+    # via datasets
+rich==15.0.0
     # via typer
-ruff==0.12.12
-    # via gradio
-safehttpx==0.1.6
+rpds-py==0.30.0
+    # via
+    #   jsonschema
+    #   referencing
+safehttpx==0.1.7
     # via gradio
-safetensors==0.6.2
+safetensors==0.7.0
     # via timm
 semantic-version==2.10.0
     # via gradio
-setuptools==80.9.0
-    # via triton
+setuptools==82.0.1 ; python_full_version >= '3.12'
+    # via torch
 shellingham==1.5.4
     # via typer
 six==1.17.0
     # via python-dateutil
-sniffio==1.3.1
-    # via anyio
-starlette==0.47.3
+sse-starlette==3.3.4
+    # via mcp
+starlette==1.0.0
     # via
     #   fastapi
     #   gradio
+    #   mcp
+    #   sse-starlette
 sympy==1.14.0
     # via torch
-timm==1.0.19
-    # via styleswin (pyproject.toml)
-tomlkit==0.13.3
+timm==1.0.26
+    # via styleswin
+tomlkit==0.14.0
     # via gradio
-torch==2.8.0
+torch==2.9.1
     # via
-    #   styleswin (pyproject.toml)
+    #   styleswin
     #   timm
     #   torchvision
-torchvision==0.23.0
+torchvision==0.24.1
     # via
-    #   styleswin (pyproject.toml)
+    #   styleswin
     #   timm
-tqdm==4.67.1
-    # via huggingface-hub
-triton==3.4.0
+tqdm==4.67.3
+    # via
+    #   datasets
+    #   huggingface-hub
+triton==3.5.1 ; platform_machine == 'x86_64' and sys_platform == 'linux'
     # via torch
-typer==0.17.3
-    # via gradio
+typer==0.24.2
+    # via
+    #   gradio
+    #   hf-gradio
+    #   huggingface-hub
 typing-extensions==4.15.0
     # via
+    #   aiosignal
     #   anyio
+    #   cryptography
     #   exceptiongroup
     #   fastapi
     #   gradio
     #   gradio-client
     #   huggingface-hub
+    #   mcp
+    #   multidict
     #   pydantic
     #   pydantic-core
+    #   pyjwt
+    #   referencing
     #   starlette
     #   torch
-    #   typer
     #   typing-inspection
     #   uvicorn
-typing-inspection==0.4.1
-    # via pydantic
-tzdata==2025.2
+typing-inspection==0.4.2
+    # via
+    #   fastapi
+    #   mcp
+    #   pydantic
+    #   pydantic-settings
+tzdata==2026.1 ; python_full_version < '3.11' or sys_platform == 'emscripten' or sys_platform == 'win32'
     # via pandas
-urllib3==2.5.0
+urllib3==2.6.3
     # via requests
-uvicorn==0.35.0
-    # via gradio
-websockets==15.0.1
-    # via gradio-client
+uvicorn==0.46.0
+    # via
+    #   gradio
+    #   mcp
+xxhash==3.6.0
+    # via datasets
+yarl==1.23.0
+    # via aiohttp