allow caching

scripts/gradio_demo.py

@@ -239,7 +239,7 @@ if not os.path.exists(save_imgs_path):
 
 
 
-def run_bite_inference(input_image, bbox=None, apply_ttopt=True):
+def run_bite_inference(input_image, bbox=None, apply_ttopt=True,  dog_name="dog_model"):
 
     with open(loss_weight_path, 'r') as j: 
         losses = json.loads(j.read())
@@ -540,7 +540,7 @@ def run_bite_inference(input_image, bbox=None, apply_ttopt=True):
                             [0, -1, 0, 0],
                             [0, 0, 1, 1],
                             [0, 0, 0, 1]])
-    result_path = os.path.join(save_imgs_path, 'dog_model')
+    result_path = os.path.join(save_imgs_path, dog_name)
     mesh.export(file_obj=result_path + '.glb')
     result_gltf = result_path + '.glb'
     return result_gltf
@@ -574,7 +574,12 @@ def run_complete_inference(img_path_or_img, crop_choice, use_ttopt):
     else:
         apply_ttopt = False
     # run barc inference
-    result_gltf = run_bite_inference(img_path_or_img, output_interm_bbox, apply_ttopt)
+    if img_path_or_img.dtype == str:
+        dog_name = os.path.basename(img_path_or_img).split(".")[0]
+    else:
+        dog_name = "dog"
+
+    result_gltf = run_bite_inference(img_path_or_img, output_interm_bbox, apply_ttopt, dog_name=dog_name)
     # add white border to image for nicer alignment
     output_interm_image_vis = np.concatenate((255*np.ones_like(output_interm_image), output_interm_image, 255*np.ones_like(output_interm_image)), axis=1)
     return [result_gltf, result_gltf, output_interm_image_vis]
@@ -659,7 +664,7 @@ demo = gr.Interface(
     examples=examples,
     thumbnail="bite_thumbnail.png",
     allow_flagging="never",
-    cache_examples=False,  # True,
+    cache_examples=True,
     examples_per_page=14,
 )