add more examples

app.py

@@ -30,9 +30,21 @@ style_image = st.file_uploader("Upload a style image")
 
 # default images
 st.write("Or you can choose from the following examples")
-col1, col2, col3 = st.columns(3)
+col1, col2, col3,col4 = st.columns(4)
 
-if col2.button("Joshua Tree"):
+if col1.button("Couple on bench"):
+  content_image = "examples/couple_on_bench.jpeg"
+  style_image = "examples/starry_night.jpeg"
+
+if col2.button("Couple Walking"):
+  content_image = "examples/couple_walking.jpeg"
+  style_image = "examples/couple_watercolor.jpeg"
+
+if col3.button("Golden Gate Bridge"):
+  content_image = "examples/golden_gate_bridge.jpeg"
+  style_image = "examples/the_great_wave.jpeg"
+
+if col4.button("Joshua Tree"):
   content_image = "examples/joshua_tree.jpeg"
   style_image = "examples/starry_night.jpeg"
 
@@ -45,13 +57,13 @@ if style_image and content_image is not None:
   # It is recommended that the style image is about 256 pixels (this size was used when training the style transfer network).
   style_image = Image.open(style_image).resize((256, 256))
 
-  output_image=perform_style_transfer(content_image, style_image)
-
   col1.header("Content Image")
   col1.image(content_image, use_column_width=True)
   col2.header("Style Image")
   col2.image(style_image, use_column_width=True)
 
+  output_image=perform_style_transfer(content_image, style_image)
+
   st.header("Output: Style transfer Image")
   st.image(output_image, use_column_width=True)