Spaces:
Sleeping
Sleeping
padmanabhbosamia
commited on
Commit
•
30d8d88
1
Parent(s):
f6a2113
Update app.py
Browse files
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 = "
|
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="
|
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()
|