multimodalart HF staff commited on
Commit
65fb376
1 Parent(s): 86427d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -227,8 +227,10 @@ def loop_outpainting(image, width, height, overlap_width, num_inference_steps, r
227
 
228
  for _ in range(num_iterations):
229
  # Generate new image
230
- result = next(infer(current_image, width, height, overlap_width, num_inference_steps, resize_option, custom_resize_size, prompt_input, alignment))
231
- new_image = result[1]
 
 
232
  image_list.append(new_image)
233
 
234
  # Use new image as input for next iteration
 
227
 
228
  for _ in range(num_iterations):
229
  # Generate new image
230
+ for step_result in infer(current_image, width, height, overlap_width, num_inference_steps, resize_option, custom_resize_size, prompt_input, alignment):
231
+ pass # Process all steps
232
+
233
+ new_image = step_result[1] # Get the final image from the last step
234
  image_list.append(new_image)
235
 
236
  # Use new image as input for next iteration