envs

app.py

@@ -181,10 +181,28 @@ image_examples = [
      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",
+# }
+
+# 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",
+# }
+
+
 
 DREAM_BOOTH = {
     'HelloObject': 'models/personalized/helloobjects_V12c.safetensors',
@@ -587,13 +605,13 @@ with block as demo:
     with gr.Row():
         def process_example(input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path): 
 
-            return input_image, prompt, drag_mode, seed, personalized
+            return input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path
   
         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],
+            outputs=[input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path],
             fn=process_example,
             run_on_click=True,
             examples_per_page=10,

requirements.txt

@@ -2,7 +2,7 @@ torch
 torchvision
 torchaudio
 transformers==4.32.1
-gradio==4.37.2
+gradio==4.0.0
 ftfy
 tensorboard
 datasets