alexnasa commited on
Commit
9fb2d90
·
verified ·
1 Parent(s): 3f81e44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -358,7 +358,11 @@ def apply_prompt(meta_data, new_prompt):
358
  pipe, _, scheduler = load_pipeline()
359
  pipe.to("cuda")
360
 
361
- caption, real_image_initial_latents, inversed_latents, _ = meta_data
 
 
 
 
362
  negative_prompt = ""
363
 
364
  inference_steps = len(inversed_latents)
 
358
  pipe, _, scheduler = load_pipeline()
359
  pipe.to("cuda")
360
 
361
+ caption, real_image_initial_latents, inversed_latents, saved_weights = meta_data
362
+ # overwrite the global so your processor can see it
363
+ global weights
364
+ weights = saved_weights
365
+
366
  negative_prompt = ""
367
 
368
  inference_steps = len(inversed_latents)