envs

app.py

@@ -288,7 +288,7 @@ class ImageConductor:
 
         self.blur_kernel = blur_kernel
 
-    @spaces.GPU(duration=200)
+    @spaces.GPU(duration=300)
     def run(self, first_frame_path, tracking_points, prompt, drag_mode, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, personalized, examples_type):
         print("Run!")
         if examples_type != "":
@@ -402,7 +402,7 @@ def preprocess_image(image, tracking_points):
     first_frame_path = os.path.join(output_dir, f"first_frame_{id}.jpg")
     image_pil.save(first_frame_path, quality=95)
     tracking_points =  gr.State([])
-    return {input_image: first_frame_path, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
+    return {input_image: first_frame_path, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points, personalized:""}
 
 
 def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.SelectData):  # SelectData is a subclass of EventData
@@ -638,7 +638,7 @@ with block:
         gr.Markdown(citation)
 
     
-    image_upload_button.upload(preprocess_image, [image_upload_button, tracking_points_var], [input_image, first_frame_path_var, tracking_points_var])
+    image_upload_button.upload(preprocess_image, [image_upload_button, tracking_points_var], [input_image, first_frame_path_var, tracking_points_var, personalized])
 
     add_drag_button.click(add_drag, tracking_points_var, tracking_points_var)