Spaces:
Sleeping
Sleeping
envs
Browse files
app.py
CHANGED
@@ -288,7 +288,7 @@ class ImageConductor:
|
|
288 |
|
289 |
self.blur_kernel = blur_kernel
|
290 |
|
291 |
-
@spaces.GPU(duration=
|
292 |
def run(self, first_frame_path, tracking_points, prompt, drag_mode, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, personalized, examples_type):
|
293 |
print("Run!")
|
294 |
if examples_type != "":
|
@@ -411,14 +411,22 @@ def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.Se
|
|
411 |
elif drag_mode=='camera':
|
412 |
color = (0, 0, 255, 255)
|
413 |
|
414 |
-
|
415 |
-
|
416 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
417 |
|
418 |
transparent_background = Image.open(first_frame_path).convert('RGBA')
|
419 |
w, h = transparent_background.size
|
420 |
transparent_layer = np.zeros((h, w, 4))
|
421 |
-
|
|
|
422 |
if len(track) > 1:
|
423 |
for i in range(len(track)-1):
|
424 |
start_point = track[i]
|
|
|
288 |
|
289 |
self.blur_kernel = blur_kernel
|
290 |
|
291 |
+
@spaces.GPU(duration=150)
|
292 |
def run(self, first_frame_path, tracking_points, prompt, drag_mode, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, personalized, examples_type):
|
293 |
print("Run!")
|
294 |
if examples_type != "":
|
|
|
411 |
elif drag_mode=='camera':
|
412 |
color = (0, 0, 255, 255)
|
413 |
|
414 |
+
if not isinstance(tracking_points ,list):
|
415 |
+
print(f"You selected {evt.value} at {evt.index} from {evt.target}")
|
416 |
+
tracking_points.value[-1].append(evt.index)
|
417 |
+
print(tracking_points.value)
|
418 |
+
tracking_points_values = tracking_points.value
|
419 |
+
else:
|
420 |
+
tracking_points[-1].append(evt.index)
|
421 |
+
tracking_points_values = tracking_points
|
422 |
+
|
423 |
+
|
424 |
|
425 |
transparent_background = Image.open(first_frame_path).convert('RGBA')
|
426 |
w, h = transparent_background.size
|
427 |
transparent_layer = np.zeros((h, w, 4))
|
428 |
+
|
429 |
+
for track in tracking_points_values:
|
430 |
if len(track) > 1:
|
431 |
for i in range(len(track)-1):
|
432 |
start_point = track[i]
|