Update app.py

app.py

@@ -171,22 +171,15 @@ def style_transfer(content_image, style_image, alpha, beta, conv1_1, conv2_1, co
     
     return final_image, intermediate_grid
 
+def load_example(content, style, output):
+        return content, style, output
 
 # Example images
 
 examples = [
-    ["assets/content_1.jpg",
-     "assets/style_1.jpg"],
-    ["assets/content_2.jpg",
-     "assets/style_2.jpg"],
-    ["assets/content_3.png",
-     "assets/style_3.jpg"],
-]
-
-examples_output = [
-    "assets/result_1.png",
-    "assets/result_2.png",
-    "assets/result_3.png"
+    ["assets/content_1.jpg", "assets/style_1.jpg", "assets/result_1.png"],
+    ["assets/content_2.jpg", "assets/style_2.jpg", "assets/result_2.png"],
+    ["assets/content_3.png", "assets/style_3.jpg", "assets/result_3.png"],
 ]
 
 #Load VGG19 model
@@ -198,7 +191,7 @@ vgg.to(device)
 
 
 # Resize example images
-resized_examples = [[resize_image(content), resize_image(style)] for content, style in examples]
+resized_examples = [[resize_image(content), resize_image(style), resize_image(output)] for content, style, output in examples]
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -245,8 +238,10 @@ with gr.Blocks() as demo:
     
     gr.Examples(
         resized_examples,
-        inputs=[content_input, style_input],
-        outputs=examples_output
+        inputs=[content_input, style_input, output_image],
+        outputs=[content_input, style_input, output_image],
+        fn=load_example,
+        cache_examples=True
     )
 
 demo.launch()