diff --git a/pyproject.toml b/pyproject.toml
index 2306c0c..f8ec849 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ dependencies = [
     "tyro",
     "fire",
     "ninja",
-    "gradio==5.17.0",
+    "gradio==5.31.0",
     "einops",
     "colorama",
     "splines",