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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -251,7 +251,7 @@ def display_images_in_rows(images_with_titles, titles):
251
  # plt.show()
252
 
253
 
254
- def image_generator(prompt = "dog", loss_function=None):
255
  images_without_loss = []
256
  images_with_loss = []
257
  if loss_function == "Yes":
@@ -280,7 +280,7 @@ def image_generator(prompt = "dog", loss_function=None):
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="dog sitting on a bench"),
284
  gr.Radio(["Yes", "No"], value="No" , label="Apply vibrance loss")],
285
  outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion using Textual Inversion", description=description)
286
  demo.launch()
 
251
  # plt.show()
252
 
253
 
254
+ def image_generator(prompt = "snoopy", loss_function=None):
255
  images_without_loss = []
256
  images_with_loss = []
257
  if loss_function == "Yes":
 
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"),
284
  gr.Radio(["Yes", "No"], value="No" , label="Apply vibrance loss")],
285
  outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion using Textual Inversion", description=description)
286
  demo.launch()