dg845 commited on
Commit
8194c9d
1 Parent(s): bd26d11

Update unidiffuser/sample_v1.py

Browse files

Fix sample function for gradio app

Files changed (1) hide show
  1. unidiffuser/sample_v1.py +2 -1
unidiffuser/sample_v1.py CHANGED
@@ -433,5 +433,6 @@ def sample(mode, prompt, image, sample_steps=50, scale=7.0, seed=None):
433
  if seed is not None:
434
  config.seed = seed
435
 
436
- evaluate(config)
 
437
 
 
433
  if seed is not None:
434
  config.seed = seed
435
 
436
+ sample_images, sample_text = evaluate(config)
437
+ return sample_images, sample_text
438