padmanabhbosamia commited on
Commit
4041358
1 Parent(s): 30d8d88

Update app.py

Browse files

Changes done on interface

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,10 +45,10 @@ style_files = ['Thumps_up.bin', 'birb_style.bin',
45
  images_without_loss = []
46
  images_with_loss = []
47
 
48
- seed_values = [8,16,50,80,128]
49
  height = 512 # default height of Stable Diffusion
50
  width = 512 # default width of Stable Diffusion
51
- num_inference_steps = 5 # Number of denoising steps
52
  guidance_scale = 7.5 # Scale for classifier-free guidance
53
  num_styles = len(style_files)
54
 
@@ -277,7 +277,7 @@ def image_generator(prompt = "snoopy", loss_function=None):
277
 
278
  return display_images_in_rows(generated_sd_images, titles)
279
 
280
- description = "Generate an image with a prompt and apply vibrance loss if you wish to. Note that the app is hosted on a cpu and it takes atleast 15 minutes for generating images without loss. Please feel free to clone the space and use it with a GPU after increase the inference steps to more than 10 for better results"
281
 
282
  demo = gr.Interface(image_generator,
283
  inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="snoopy sitting on a bench"),
 
45
  images_without_loss = []
46
  images_with_loss = []
47
 
48
+ seed_values = [10,12,18,30,32]
49
  height = 512 # default height of Stable Diffusion
50
  width = 512 # default width of Stable Diffusion
51
+ num_inference_steps = 30 # Number of denoising steps
52
  guidance_scale = 7.5 # Scale for classifier-free guidance
53
  num_styles = len(style_files)
54
 
 
277
 
278
  return display_images_in_rows(generated_sd_images, titles)
279
 
280
+ description = "Generate Image from Prompt.Time Taken for Image is around 20 minutes. Please run it on GPU for better performance"
281
 
282
  demo = gr.Interface(image_generator,
283
  inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="snoopy sitting on a bench"),