example_list

app.py

@@ -3,6 +3,7 @@ import ot
 import numpy as np
 import gradio as gr
 
+
 def preprocess(content_img,style_img):
 
     content_img = Image.fromarray(content_img.astype("uint8"), "RGB")
@@ -34,21 +35,11 @@ def preprocess(content_img,style_img):
     return transferred_img
 
 
-# img1_file = 'huggingface/otd/gohho.png'
-img1_file = 'gohho.png'
-
-img1 = Image.open(img1_file)
-# img2_file = 'huggingface/otd/sea.png'
-img2_file = 'sea.png'
-img2 = Image.open(img2_file)
-
+content_file = 'gohho.png'
+style_file = 'sea.png'
 
-gr_img = gr.Image(np.array(img1))
-gr_img2 = gr.Image(np.array(img2))
-
-
-# gr_img = gr.inputs.Image()
-# gr_img2 = gr.inputs.Image()
+gr_img = gr.inputs.Image()
+gr_img2 = gr.inputs.Image()
 
 interface = gr.Interface(
     fn=preprocess,
@@ -57,6 +48,9 @@ interface = gr.Interface(
         gr_img2
     ],
     outputs="image",
+    examples=[
+        [content_file, style_file],
+    ],
     title="最適輸送距離による色相変換",
     description="content画像と，style画像を入力してください",
 )