fffiloni commited on
Commit
5c8ebde
·
1 Parent(s): 1127fc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- i = Image.open(i).convert("RGB")
142
- i = np.array(i)
143
- output_frame = edit(i,
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,