code update

app.py

@@ -272,7 +272,6 @@ def image_generator(prompt = "dog", loss_function=None):
 
 # Create a wrapper function for show_misclassified_images()
 def image_generator_wrapper(prompt = "dog", loss_function=None):
-    prompt = string(prompt)
     if loss_function == "Yes":
       loss_function = vibrance_loss
     else:
@@ -283,7 +282,7 @@ def image_generator_wrapper(prompt = "dog", loss_function=None):
 description = "Generate an image with a prompt and apply loss if you wish to"
 
 demo = gr.Interface(image_generator_wrapper,
-                    inputs=[gr.Textbox(label="Enter prompt for generating", type="text", default="dog sitting on a bench"),
+                    inputs=[gr.Textbox(label="Enter prompt for generating", type="text", value="dog sitting on a bench"),
                             gr.Radio(["Yes", "No"], value="No"  , label="Apply vibrance loss")],
                     outputs=gr.Plot(label="Outputs"), title = "Stable Diffusion", description=description)
 demo.launch()