diff --git a/mnist-classifaction/deploy_model/deploy.py b/mnist-classifaction/deploy_model/deploy.py index 3d382de..8dea3b1 100644 --- 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 index 0719d0d..4b5f9ca 100644 --- 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", "/"))