envs

app.py

@@ -116,87 +116,32 @@ os.system(f'mv models/personalized/TUSUN.safetensors?download=true models/person
 
 # - - - - - examples  - - - - -  #
 
-# image_examples = [
-#     ["__asset__/images/object/turtle-1.jpg", 
-#      "a sea turtle gracefully swimming over a coral reef in the clear blue ocean.", 
-#      "object",
-#      11318446767408804497,
-#      "",
-#      json.load(open("__asset__/trajs/object/turtle-1.json")), 
-#      "__asset__/images/object/turtle-1.jpg",
-#      ],
-    
-#     ["__asset__/images/object/rose-1.jpg", 
-#      "a red rose engulfed in flames.", 
-#      "object",
-#      6854275249656120509,
-#      "",
-#      json.load(open("__asset__/trajs/object/rose-1.json")), 
-#      "__asset__/images/object/rose-1.jpg", 
-#      ],
-    
-#     ["__asset__/images/object/jellyfish-1.jpg", 
-#      "intricate detailing,photorealism,hyperrealistic, glowing jellyfish mushroom, flying, starry sky, bokeh, golden ratio composition.", 
-#      "object",
-#      17966188172968903484,
-#      "HelloObject",
-#      json.load(open("__asset__/trajs/object/jellyfish-1.json")), 
-#      "__asset__/images/object/jellyfish-1.jpg",
-#      ],
-    
-    
-#     ["__asset__/images/camera/lush-1.jpg", 
-#      "detailed craftsmanship, photorealism, hyperrealistic, roaring waterfall, misty spray, lush greenery, vibrant rainbow, golden ratio composition.", 
-#      "camera",
-#      7970487946960948963,
-#      "HelloObject",
-#      json.load(open("__asset__/trajs/camera/lush-1.json")),
-#      "__asset__/images/camera/lush-1.jpg", 
-#      ],
-    
-#     ["__asset__/images/camera/tusun-1.jpg", 
-#      "tusuncub with its mouth open, blurry, open mouth, fangs, photo background, looking at viewer, tongue, full body, solo, cute and lovely, Beautiful and realistic eye details, perfect anatomy, Nonsense, pure background, Centered-Shot, realistic photo, photograph, 4k, hyper detailed, DSLR, 24 Megapixels, 8mm Lens, Full Frame, film grain, Global Illumination, studio Lighting, Award Winning Photography, diffuse reflection, ray tracing.", 
-#      "camera",
-#      996953226890228361,
-#      "TUSUN",
-#      json.load(open("__asset__/trajs/camera/tusun-1.json")),
-#      "__asset__/images/camera/tusun-1.jpg",
-#      ],
-    
-#     ["__asset__/images/camera/painting-1.jpg", 
-#      "A oil painting.", 
-#      "camera",
-#      16867854766769816385,
-#      "",
-#      json.load(open("__asset__/trajs/camera/painting-1.json")),
-#      "__asset__/images/camera/painting-1.jpg", 
-#      ],
-    
-# ]
-
 image_examples = [
     ["__asset__/images/object/turtle-1.jpg", 
      "a sea turtle gracefully swimming over a coral reef in the clear blue ocean.", 
      "object",
      11318446767408804497,
-    #  json.load(open("__asset__/trajs/object/turtle-1.json")), 
-     "__asset__/trajs/object/turtle-1.json",
+     "",
+     json.load(open("__asset__/trajs/object/turtle-1.json")), 
+     "__asset__/images/object/turtle-1.jpg",
      ],
     
     ["__asset__/images/object/rose-1.jpg", 
      "a red rose engulfed in flames.", 
      "object",
      6854275249656120509,
-    #  json.load(open("__asset__/trajs/object/rose-1.json")), 
-    "__asset__/trajs/object/turtle-1.json",
+     "",
+     json.load(open("__asset__/trajs/object/rose-1.json")), 
+     "__asset__/images/object/rose-1.jpg", 
      ],
     
     ["__asset__/images/object/jellyfish-1.jpg", 
      "intricate detailing,photorealism,hyperrealistic, glowing jellyfish mushroom, flying, starry sky, bokeh, golden ratio composition.", 
      "object",
      17966188172968903484,
-    #  json.load(open("__asset__/trajs/object/jellyfish-1.json")), 
-     "__asset__/trajs/object/turtle-1.json",
+     "HelloObject",
+     json.load(open("__asset__/trajs/object/jellyfish-1.json")), 
+     "__asset__/images/object/jellyfish-1.jpg",
      ],
     
     
@@ -204,29 +149,33 @@ image_examples = [
      "detailed craftsmanship, photorealism, hyperrealistic, roaring waterfall, misty spray, lush greenery, vibrant rainbow, golden ratio composition.", 
      "camera",
      7970487946960948963,
-    #  json.load(open("__asset__/trajs/camera/lush-1.json")),
-    "__asset__/trajs/object/turtle-1.json",
+     "HelloObject",
+     json.load(open("__asset__/trajs/camera/lush-1.json")),
+     "__asset__/images/camera/lush-1.jpg", 
      ],
     
     ["__asset__/images/camera/tusun-1.jpg", 
      "tusuncub with its mouth open, blurry, open mouth, fangs, photo background, looking at viewer, tongue, full body, solo, cute and lovely, Beautiful and realistic eye details, perfect anatomy, Nonsense, pure background, Centered-Shot, realistic photo, photograph, 4k, hyper detailed, DSLR, 24 Megapixels, 8mm Lens, Full Frame, film grain, Global Illumination, studio Lighting, Award Winning Photography, diffuse reflection, ray tracing.", 
      "camera",
      996953226890228361,
-    #  json.load(open("__asset__/trajs/camera/tusun-1.json")),
-    "__asset__/trajs/object/turtle-1.json",
+     "TUSUN",
+     json.load(open("__asset__/trajs/camera/tusun-1.json")),
+     "__asset__/images/camera/tusun-1.jpg",
      ],
     
     ["__asset__/images/camera/painting-1.jpg", 
      "A oil painting.", 
      "camera",
      16867854766769816385,
-    #  json.load(open("__asset__/trajs/camera/painting-1.json")),
-    "__asset__/trajs/object/turtle-1.json",
+     "",
+     json.load(open("__asset__/trajs/camera/painting-1.json")),
+     "__asset__/images/camera/painting-1.jpg", 
      ],
     
 ]
 
 
+
 DREAM_BOOTH = {
     'HelloObject': 'models/personalized/helloobjects_V12c.safetensors',
 }
@@ -554,8 +503,8 @@ with block as demo:
                                         width=384,
                                         model_length=16
                                         )
-    first_frame_path = gr.State(value=None,label="image")
-    tracking_points = gr.State([], label="points")
+    first_frame_path = gr.State(value=None,label="")
+    tracking_points = gr.State([], label="")
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -626,34 +575,22 @@ 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, tracking_points, first_frame_path): 
 
-        #     return input_image, prompt, drag_mode, seed, personalized, tracking_points, first_frame_path
-        def process_example(input_image, prompt, drag_mode, seed, tracking_points): 
-            return input_image, prompt, drag_mode, seed, tracking_points
-        
-        # 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],
-        #     fn=process_example,
-        #     run_on_click=True,
-        #     examples_per_page=10,
-        #     cache_examples=False,
-        # )
-        
+            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, tracking_points],
-            outputs=[input_image, prompt, drag_mode, seed, tracking_points],
+            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],
             fn=process_example,
             run_on_click=True,
-            examples_per_page=20,
+            examples_per_page=10,
             cache_examples=False,
         )
         
+        
     with gr.Row():
         gr.Markdown(citation)