Manjushri commited on
Commit
8c45ca5
1 Parent(s): 69149d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -365,7 +365,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
365
  torch.cuda.empty_cache()
366
  refined = refiner(Prompt, negative_prompt=negative_prompt, image=image, denoising_start=high_noise_frac).images[0]
367
  torch.cuda.empty_cache()
368
-
369
  return image
370
 
371
  gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Anime', 'Disney', 'StoryBook', 'SemiReal', 'Animagine XL 3.0', 'SDXL 1.0', 'FusionXL', 'SDXL-Turbo'], value='PhotoReal', label='Choose Model'),
 
365
  torch.cuda.empty_cache()
366
  refined = refiner(Prompt, negative_prompt=negative_prompt, image=image, denoising_start=high_noise_frac).images[0]
367
  torch.cuda.empty_cache()
368
+ return refined
369
  return image
370
 
371
  gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Anime', 'Disney', 'StoryBook', 'SemiReal', 'Animagine XL 3.0', 'SDXL 1.0', 'FusionXL', 'SDXL-Turbo'], value='PhotoReal', label='Choose Model'),