examples

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from paths import *
-
+import os
 from vision_tower import DINOv2_MLP
 from transformers import AutoImageProcessor
 import torch
@@ -51,9 +51,14 @@ def infer_func(img, do_rm_bkg, do_infer_aug):
     # axis_model = "axis.obj"
     return [res_img, round(float(angles[0]), 2), round(float(angles[1]), 2), round(float(angles[2]), 2), round(float(angles[3]), 2)]
 
+example_files = os.listdir('examples')
+example_files.sort()
+example_files = [[os.path.join('examples', filename), None, None] for filename in example_files]
+
 server = gr.Interface(
     flagging_mode='never',
-    fn=infer_func, 
+    fn=infer_func,
+    examples=example_files,
     inputs=[
         gr.Image(height=512, width=512, label="upload your image"),
         gr.Checkbox(label="Remove Background", value=True),