Update app.py
Browse files
app.py
CHANGED
@@ -131,16 +131,16 @@ def infer(video_in, do_inversion, wts, zs,
|
|
131 |
result_frames = []
|
132 |
print("set stop frames to: " + str(n_frame))
|
133 |
|
134 |
-
for i in frames_list[0:int(n_frame)]:
|
135 |
#mmpose_frame = get_mmpose_filter(i)
|
136 |
|
137 |
#seed operations
|
138 |
seed = randomize_seed_fn(seed, randomize_seed)
|
139 |
do_inversion = True
|
140 |
|
141 |
-
|
142 |
-
|
143 |
-
output_frame = edit(
|
144 |
do_inversion, wts, zs,
|
145 |
src_prompt,
|
146 |
tar_prompt,
|
|
|
131 |
result_frames = []
|
132 |
print("set stop frames to: " + str(n_frame))
|
133 |
|
134 |
+
for i, image in enumerate(frames_list[0:int(n_frame)]):
|
135 |
#mmpose_frame = get_mmpose_filter(i)
|
136 |
|
137 |
#seed operations
|
138 |
seed = randomize_seed_fn(seed, randomize_seed)
|
139 |
do_inversion = True
|
140 |
|
141 |
+
image = Image.open(image).convert("RGB")
|
142 |
+
image = np.array(image)
|
143 |
+
output_frame = edit(image,
|
144 |
do_inversion, wts, zs,
|
145 |
src_prompt,
|
146 |
tar_prompt,
|