Customizing UI components

app.py

@@ -17,10 +17,11 @@ def perform_style_transfer(content_image, style_image):
     return Image.fromarray(np.uint8(stylized_image[0] * 255))
 
 
-content_image_input = gr.inputs.Image()
-style_image_input = gr.inputs.Image(shape=(256, 256))
+content_image_input = gr.inputs.Image(label="Content Image")
+style_image_input = gr.inputs.Image(shape=(256, 256), label="Style Image")
 
 app_interface = gr.Interface(fn=perform_style_transfer,
-                  inputs=[content_image_input, style_image_input],
-                  outputs="image")
+                             inputs=[content_image_input, style_image_input],
+                             outputs="image",
+                             title="Fast Neural Style Transfer")
 app_interface.launch()