nit: update root path for gradio

truefoundry · Feb 19, 2024 · e5f583a · e5f583a
1 parent 3ec1241
commit e5f583a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mnist-classifaction/deploy_model/deploy.py b/mnist-classifaction/deploy_model/deploy.py
@@ -44,6 +44,6 @@
                 download_path_env_variable="MODEL_DOWNLOAD_PATH",
             )
         ]
-    ),
+    )
 )
 service.deploy(workspace_fqn=args.workspace_fqn)
diff --git a/mnist-classifaction/deploy_model/gradio_demo.py b/mnist-classifaction/deploy_model/gradio_demo.py
@@ -17,4 +17,4 @@ def get_inference(img_arr):
     examples=[["sample_images/0.jpg"], ["sample_images/1.jpg"]],
 )
 
-interface.launch(server_name="0.0.0.0", server_port=8000)
+interface.launch(server_name="0.0.0.0", server_port=8000, root_path=os.environ.get("TFY_SERVICE_ROOT_PATH", "/"))
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,6 +44,6 @@ @@
                     download_path_env_variable="MODEL_DOWNLOAD_PATH",
                 )
             ]
-        ),
+        )
     )
     service.deploy(workspace_fqn=args.workspace_fqn)