code update

app.py

@@ -283,7 +283,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="str", default="dog sitting on a bench"),
+                    inputs=[gr.Textbox(label="Enter prompt for generating", type="text", default="dog sitting on a bench"),
                             gr.Radio(["Yes", "No"], value="No"  , label="Apply vibrance loss")],
-                    outputs=gr.Plot(), title = "Stable Diffusion", description=description)
+                    outputs=gr.Plot(label="Outputs"), title = "Stable Diffusion", description=description)
 demo.launch()