linoyts HF Staff commited on
Commit
5ad8749
·
verified ·
1 Parent(s): 15e6d77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ examples = [
93
  ]
94
 
95
  def generate_examples(input_image, prompt):
96
- return process(input_image, prompt)
97
 
98
 
99
  @torch.no_grad()
 
93
  ]
94
 
95
  def generate_examples(input_image, prompt):
96
+ yield process(input_image, prompt)
97
 
98
 
99
  @torch.no_grad()