Update app.py

app.py

@@ -47,6 +47,14 @@ def fx5():
 def fx6():
     return ht_text
     
+def FX7(input_path,input_path2):
+  input_img = load_img(input_path)
+  input_img2 = load_img(input_path2)
+  hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
+  stylized_image = hub_model(tf.constant(input_img), tf.constant(input_img2))[0]
+  pre_img = tensor_to_image(stylized_image)
+  return pre_img
+    
 with demo:
     gr.Markdown(
     """
@@ -76,6 +84,10 @@ with demo:
         with gr.TabItem("Excel365测试"):
             lk_button = gr.Button("Excel调用>>")
             lk_output=gr.HTML(label="在线版演示",show_label=True)
+        with gr.TabItem("风格迁移"):
+            qy_input=[gr.Image(label="待转换图片",type="filepath"),gr.Image(label="风格图片",type="filepath")]
+            qy_output=gr.Image(type="auto", label="输出")
+            qy_button = gr.Button("开始迁移>>")
             
     tj_button.click(fx1, inputs=text_input, outputs=text_output)
     cs_button.click(fx2, inputs=val_input, outputs=json_output)
@@ -83,5 +95,6 @@ with demo:
     tts_button.click(fx4, inputs=TS_input, outputs=audio_output)
     pva_button.click(fx5, inputs=[],outputs=pva_output)
     lk_button.click(fx6, inputs=[],outputs=lk_output)
+    qy_button.click(fx7,examples=[[demo_path,demo_path2],],inputs=qy_input,outputs=qy_output,theme="dark-seafoam")
 
 demo.launch()