envs

app.py

@@ -131,9 +131,6 @@ image_examples = [
      "object",
      11318446767408804497,
      "",
-     json.load(open("__asset__/trajs/object/turtle-1.json")), 
-     "__asset__/images/object/turtle-1.jpg",
-     ""
      ],
     
     ["__asset__/images/object/rose-1.jpg", 
@@ -141,8 +138,6 @@ image_examples = [
      "object",
      6854275249656120509,
      "",
-     json.load(open("__asset__/trajs/object/rose-1.json")), 
-     "__asset__/images/object/rose-1.jpg", 
      ],
     
     ["__asset__/images/object/jellyfish-1.jpg", 
@@ -150,8 +145,6 @@ image_examples = [
      "object",
      17966188172968903484,
      "HelloObject",
-     json.load(open("__asset__/trajs/object/jellyfish-1.json")), 
-     "__asset__/images/object/jellyfish-1.jpg",
      ],
     
     
@@ -160,8 +153,6 @@ image_examples = [
      "camera",
      7970487946960948963,
      "HelloObject",
-     json.load(open("__asset__/trajs/camera/lush-1.json")),
-     "__asset__/images/camera/lush-1.jpg", 
      ],
     
     ["__asset__/images/camera/tusun-1.jpg", 
@@ -169,8 +160,6 @@ image_examples = [
      "camera",
      996953226890228361,
      "TUSUN",
-     json.load(open("__asset__/trajs/camera/tusun-1.json")),
-     "__asset__/images/camera/tusun-1.jpg",
      ],
     
     ["__asset__/images/camera/painting-1.jpg", 
@@ -178,29 +167,27 @@ image_examples = [
      "camera",
      16867854766769816385,
      "",
-     json.load(open("__asset__/trajs/camera/painting-1.json")),
-     "__asset__/images/camera/painting-1.jpg", 
      ],
 ]
 
 
-# POINTS = {
-#     'turtle': "__asset__/trajs/object/turtle-1.json",
-#     'rose': "__asset__/trajs/object/rose-1.json",
-#     'jellyfish': "__asset__/trajs/object/jellyfish-1.json",
-#     'lsuh': "__asset__/trajs/camera/lush-1.json",
-#     'tusun': "__asset__/trajs/camera/tusun-1.json",
-#     'painting': "__asset__/trajs/camera/painting-1.json",
-# }
+POINTS = {
+    'turtle': "__asset__/trajs/object/turtle-1.json",
+    'rose': "__asset__/trajs/object/rose-1.json",
+    'jellyfish': "__asset__/trajs/object/jellyfish-1.json",
+    'lsuh': "__asset__/trajs/camera/lush-1.json",
+    'tusun': "__asset__/trajs/camera/tusun-1.json",
+    'painting': "__asset__/trajs/camera/painting-1.json",
+}
 
-# IMAGE_PATH = {
-#     'turtle': "__asset__/images/object/turtle-1.jpg",
-#     'rose': "__asset__/images/object/rose-1.jpg",
-#     'jellyfish': "__asset__/images/object/jellyfish-1.jpg",
-#     'lsuh': "__asset__/images/camera/lush-1.jpg",
-#     'tusun': "__asset__/images/camera/tusun-1.jpg",
-#     'painting': "__asset__/images/camera/painting-1.jpg",
-# }
+IMAGE_PATH = {
+    'turtle': "__asset__/images/object/turtle-1.jpg",
+    'rose': "__asset__/images/object/rose-1.jpg",
+    'jellyfish': "__asset__/images/object/jellyfish-1.jpg",
+    'lsuh': "__asset__/images/camera/lush-1.jpg",
+    'tusun': "__asset__/images/camera/tusun-1.jpg",
+    'painting': "__asset__/images/camera/painting-1.jpg",
+}
 
 
 
@@ -531,8 +518,8 @@ with block as demo:
                                         width=384,
                                         model_length=16
                                         )
-    first_frame_path = gr.State()
-    tracking_points = gr.State([])
+    first_frame_path = gr.State(label='')
+    tracking_points = gr.State([], label='')
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -603,15 +590,15 @@ with block as demo:
             
             
     with gr.Row():
-        def process_example(input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path): 
+        def process_example(input_image, prompt, drag_mode, seed, personalized, ): 
 
-            return input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path
+            return input_image, prompt, drag_mode, seed, personalized
   
         example = gr.Examples(
             label="Input Example",
             examples=image_examples,
-            inputs=[input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path],
-            outputs=[input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path],
+            inputs=[input_image, prompt, drag_mode, seed, personalized],
+            outputs=[input_image, prompt, drag_mode, seed, personalized],
             fn=process_example,
             run_on_click=True,
             examples_per_page=10,