From 351c1f06b23aaabe4f6ba1079966f4bdff2b9ebc Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Mon, 14 Feb 2022 16:05:05 +0800
Subject: [PATCH] Update anime-transfer

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 .gitattributes           | 55 ++++++++++++++++++++++------------------
 anime_transfer_gradio.py | 11 ++++----
 2 files changed, 37 insertions(+), 29 deletions(-)

diff --git a/.gitattributes b/.gitattributes
index ad2c207..ab2832a 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -1,28 +1,35 @@
+# .gitattributes
 
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
+# Source files
+# ============
+*.pxd    text diff=python
+*.py     text diff=python
+*.py3    text diff=python
+*.pyw    text diff=python
+*.pyx    text diff=python
+*.pyz    text diff=python
+*.pyi    text diff=python
+
+# Binary files
+# ============
+*.db     binary
+*.p      binary
+*.pkl    binary
+*.pickle binary
+*.pyc    binary export-ignore
+*.pyo    binary export-ignore
+*.pyd    binary
+
+# Jupyter notebook
+*.ipynb  text
+
+# Model files
 *.bin.* filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text 
-*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zstandard filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.tar.gz filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
\ No newline at end of file
diff --git a/anime_transfer_gradio.py b/anime_transfer_gradio.py
index 5ddc40d..a7a29b6 100644
--- a/anime_transfer_gradio.py
+++ b/anime_transfer_gradio.py
@@ -6,6 +6,9 @@ from pathlib import Path
 from towhee.operator import Operator
 from towhee import pipeline
 
+path = str(Path.cwd() / 'test.jpg')
+
+
 class AnimeTransferGradio(Operator):
     """
     AnimeTransferGradio operator
@@ -13,11 +16,13 @@ class AnimeTransferGradio(Operator):
     def __init__(self) -> None:
         super().__init__()
 
-    def __call__(self, source: str = 'upload') -> None:
+    def __call__(self, source: str = 'upload') -> NamedTuple('Outputs', [('img_path', str)]):
         interface = gradio.Interface(self.trans_img, [gradio.inputs.Image(type="pil", source=source),
                                                       gradio.inputs.Radio(["celeba", "facepaintv1","facepaintv2", "hayao", "paprika", 'shinkai'])],
                                      gradio.outputs.Image(type="pil"), allow_flagging='never', allow_screenshot=False)
         interface.launch(enable_queue=True)
+        Outputs = NamedTuple('Outputs', [('img_path', str)])
+        return Outputs(path)
 
     @staticmethod
     def trans_img(input, version):
@@ -26,10 +31,6 @@ class AnimeTransferGradio(Operator):
         # Resizing the image while keeping aspect ratio.
         size = (512, 512)
         input.thumbnail(size, Image.ANTIALIAS)
-        # Saving image to file for input. Very low chance of concurrent file saves during the time
-        # between saving and taking first step of pipeline, so avoiding locks for now. In addition,
-        # current gradio is set to queue so there will never be parallel runs for this.
-        path = str(Path.cwd() / 'test.jpg')
         input.save(path)
 
         if version == 'celeba':