code update

app.py

@@ -48,7 +48,7 @@ images_with_loss = []
 seed_values = [8,16,50,80,128]
 height = 512                        # default height of Stable Diffusion
 width = 512                         # default width of Stable Diffusion
-num_inference_steps = 10            # Number of denoising steps
+num_inference_steps = 1            # Number of denoising steps
 guidance_scale = 7.5                # Scale for classifier-free guidance
 num_styles = len(style_files)
 
@@ -286,10 +286,10 @@ def image_generator_wrapper(prompt = "dog", loss_function=None):
 
   return image_generator(prompt, loss_function)
 
-description = "Generate an image with a prompt and apply vibrance loss if you wish to"
+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 the inference steps are reduced to 1 to produce results faster at the cost of accuracy of generated images."
 
 demo = gr.Interface(image_generator,
                     inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="dog sitting on a bench"),
                             gr.Radio(["Yes", "No"], value="No"  , label="Apply vibrance loss")],
-                    outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion", description=description)
+                    outputs=gr.Plot(label="Generated Images"), title = "Stable Diffusion using Textual Inversion", description=description)
 demo.launch()