darshanjani commited on
Commit
c2ebf37
·
1 Parent(s): 96a182b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -312,11 +312,11 @@ def image_generator_wrapper(prompt = "dog", loss_function=None):
312
 
313
  return image_generator(prompt, loss_function)
314
 
315
- description = 'Stable Diffusion is a generative artificial intelligence (generative AI) model that produces unique photorealistic images from text and image prompts.'
316
  title = 'Image Generation using Stable Diffusion'
317
 
318
  demo = gr.Interface(image_generator_wrapper,
319
- inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="astronaut riding a cycle"),
320
- gr.Radio(["Yes", "No"], value="No" , label="Apply Contrast Loss")],
321
  outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion", description=description)
322
  demo.launch()
 
312
 
313
  return image_generator(prompt, loss_function)
314
 
315
+ description = '(Team Project EE267) Stable Diffusion is a generative artificial intelligence (generative AI) model that produces unique photorealistic images from text and image prompts.'
316
  title = 'Image Generation using Stable Diffusion'
317
 
318
  demo = gr.Interface(image_generator_wrapper,
319
+ inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="a dog walking on moon"),
320
+ gr.Radio(["Yes", "No"], value="No" , label="Custom Loss Function")],
321
  outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion", description=description)
322
  demo.launch()