Update app.py

app.py

@@ -67,6 +67,21 @@ def fx6(img, lang):
     im_show = Image.fromarray(im_show)
     im_show.save('result.jpg')
     return 'result.jpg'
+
+model2 = torch.hub.load(
+    "AK391/animegan2-pytorch:main",
+    "generator",
+    pretrained=True,
+    device="cuda",
+    progress=False
+)
+face2paint = torch.hub.load(
+    'AK391/animegan2-pytorch:main', 'face2paint', 
+    size=512, device="cuda",side_by_side=False
+)
+def fx7(img):
+    out = face2paint(model2, img)
+    return out
     
     
 with demo:
@@ -101,6 +116,10 @@ with demo:
             ocr_input=[gr.Image(type='file', label='请提供你需要识别的照片'),gr.Dropdown(choices=['中文','英文','法文','德文','韩文','日文'], type="value", default='中文', label='请选择需要识别的语言')]
             ocr_output=gr.Image(type='file', label='识别结果')
             ocr_button = gr.Button("开始ocr识别")
+         with gr.TabItem("风格迁移"):
+            qy_input=gr.Image(type="pil",label="请选择需要风格迁移的照片",show_label=True)
+            qy_output=gr.Image(type="pil",label="输出迁移风格结果",show_label=True)
+            qy_button = gr.Button("开始调用模型")
             
     tj_button.click(fx1, inputs=text_input, outputs=text_output)
     cs_button.click(fx2, inputs=val_input, outputs=json_output,api_name="ghqj")
@@ -108,5 +127,6 @@ with demo:
     tts_button.click(fx4, inputs=TS_input, outputs=audio_output)
     pva_button.click(fx5, inputs=[],outputs=pva_output)
     ocr_button.click(fx6,inputs=ocr_input,outputs=ocr_output)
+    qy_button.click(fx7,inputs=qy_input,outputs=qy_output)
 
 demo.launch(enable_queue=True)