CiaraRowles commited on
Commit
8d30381
·
verified ·
1 Parent(s): 0f33a11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -40,8 +40,9 @@ def generate_image(images, prompt, negative_prompt,stylecode, progress=gr.Progr
40
 
41
  @spaces.GPU(enable_queue=True)
42
  def make_stylecode_gui (images, progress=gr.Progress(track_tqdm=True)):
43
- code = make_stylecode(ip_ckpt,images)
44
- return code
 
45
 
46
 
47
  def swap_to_gallery(images):
 
40
 
41
  @spaces.GPU(enable_queue=True)
42
  def make_stylecode_gui (images, progress=gr.Progress(track_tqdm=True)):
43
+ yield None
44
+ code = make_stylecode(ip_ckpt,image_path=None,image=images)
45
+ yield code
46
 
47
 
48
  def swap_to_gallery(images):