code update

app.py

@@ -277,16 +277,7 @@ def image_generator(prompt = "dog", loss_function=None):
 
   return display_images_in_rows(generated_sd_images, titles)
 
-# Create a wrapper function for show_misclassified_images()
-def image_generator_wrapper(prompt = "dog", loss_function=None):
-  if loss_function == "Yes":
-    loss_function = vibrance_loss
-  else:
-    loss_function = None
-
-  return image_generator(prompt, loss_function)
-
-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."
+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"
 
 demo = gr.Interface(image_generator,
                     inputs=[gr.Textbox(label="Enter prompt for generation", type="text", value="dog sitting on a bench"),