diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
index c6312b09..791e6f84 100644
--- a/cpp/CMakeLists.txt
+++ b/cpp/CMakeLists.txt
@@ -27,8 +27,8 @@ include(FetchContent)
 
 FetchContent_Declare(
   iree
-  GIT_REPOSITORY https://github.com/iree-org/iree.git
-  GIT_TAG c4624870a8de85224c3f02f4ea8e26837cd99e2f # 2022-08-12
+  GIT_REPOSITORY https://github.com/nod-ai/shark-runtime.git
+  GIT_TAG f5d84869cf4b745cc2461396978886e503861a88
   GIT_SUBMODULES_RECURSE OFF
   GIT_SHALLOW OFF
   GIT_PROGRESS ON
diff --git a/cpp/vulkan_gui/vulkan_inference_gui.cc b/cpp/vulkan_gui/vulkan_inference_gui.cc
index dff106fb..2cdd10c8 100644
--- a/cpp/vulkan_gui/vulkan_inference_gui.cc
+++ b/cpp/vulkan_gui/vulkan_inference_gui.cc
@@ -895,13 +895,13 @@ extern "C" int iree_main(int argc, char** argv) {
   */
 
   iree_file_toc_t module_file_toc;
-  load_file("amd-resnet50.vmfb", &module_file_toc.data, &module_file_toc.size);
+  load_file("resnet50_tf.vmfb", &module_file_toc.data, &module_file_toc.size);
   fprintf(stdout, "module size: %lu\n", module_file_toc.size);
 
   static float input_res50[224*224*3];
   static float output_res50[1000];
 
-  char filename[] = "dog_imagenet.jpg";
+  char filename[] = "vulkan_gui/dog_imagenet.jpg";
   fprintf(stdout, "loading: %s\n", filename);
   int x,y,n;
   unsigned char *image_raw = stbi_load(filename, &x, &y, &n, 0);
@@ -954,7 +954,7 @@ extern "C" int iree_main(int argc, char** argv) {
 
   // Lookup the entry point function.
   iree_vm_function_t main_function;
-  const char kMainFunctionName[] = "module.predict";
+  const char kMainFunctionName[] = "module.forward";
   IREE_CHECK_OK(iree_vm_context_resolve_function(
       iree_context,
       iree_string_view_t{kMainFunctionName, sizeof(kMainFunctionName) - 1},