Add application file

.gitignore

@@ -0,0 +1,4 @@
+.idea
+PAMA/
+samples/
+flagged/

README.md

@@ -7,7 +7,7 @@ sdk: gradio
 sdk_version: 3.18.0
 app_file: app.py
 pinned: false
-license: mit
+license: MIT
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Based on [https://huggingface.co/spaces/akhaliq/PAMA](https://huggingface.co/spaces/akhaliq/PAMA).

app.py

@@ -0,0 +1,55 @@
+import os
+
+import gradio as gr
+
+os.system("bash install.sh")
+
+
+def inference(content: str, style: str) -> str:
+    os.system("cd PAMA && python main.py eval --content " + content + " --style " + style)
+    return "PAMA/ics.jpg"
+
+
+title = "PAMA"
+description = (
+    "<p style='text-align: center'>"
+    "PAMA: Consistent Style Transfer. </br> To use it, simply upload "
+    "your images, or click one of the examples to load them.</p>"
+)
+
+article = (
+    "<p style='text-align: center'>"
+    "<a href='https://arxiv.org/abs/2201.02233' target='_blank'>"
+    "Consistent Style Transfer</a> | "
+    "<a href='https://github.com/luoxuan-cs/PAMA' target='_blank'>"
+    "Github Repo</a></p>"
+)
+
+examples = [
+    ['samples/800px-Hoover_Tower_Stanford_January_2013.jpg',
+     'samples/Leger_railway_crossing.jpg'],
+    ['samples/800px-Robert_Downey_Jr_2014_Comic_Con_(cropped).jpg',
+     'samples/1280px-El_Tres_de_Mayo,_by_Francisco_de_Goya,_from_Prado_thin_black_margin.jpg'],
+    ['samples/800px-Taylor_Swift_2_-_2019_by_Glenn_Francis_(cropped)_3.jpg',
+     'samples/WLANL_-_andrevanb_-_Falaises_près_de_Pourville,_Claude_Monet,_1882.jpg'],
+]
+
+
+def build_interface() -> gr.Interface:
+    return gr.Interface(
+        inference,
+        inputs=[
+            gr.Image(type="filepath", label="Content"),
+            gr.Image(type="filepath", label="Style"),
+        ],
+        outputs=gr.Image(type="filepath", label="Output"),
+        title=title,
+        description=description,
+        article=article,
+        examples=examples,
+    )
+
+
+if __name__ == "__main__":
+    interface = build_interface()
+    interface.launch()

install.sh

@@ -0,0 +1,21 @@
+echo "--- Cloning PAMA git repository"
+git clone https://github.com/AK391/PAMA.git
+
+echo "--- Download sample images"
+mkdir -p samples && cd samples
+wget "https://upload.wikimedia.org/wikipedia/commons/thumb/1/1e/Hoover_Tower_Stanford_January_2013.jpg/800px-Hoover_Tower_Stanford_January_2013.jpg"
+wget "https://upload.wikimedia.org/wikipedia/commons/thumb/e/ea/Van_Gogh_-_Starry_Night_-_Google_Art_Project.jpg/1920px-Van_Gogh_-_Starry_Night_-_Google_Art_Project.jpg"
+wget "https://upload.wikimedia.org/wikipedia/commons/8/8e/WLANL_-_andrevanb_-_Falaises_pr%C3%A8s_de_Pourville%2C_Claude_Monet%2C_1882.jpg"
+wget "https://upload.wikimedia.org/wikipedia/en/f/f7/Leger_railway_crossing.jpg"
+wget "https://upload.wikimedia.org/wikipedia/commons/thumb/f/fb/Taylor_Swift_2_-_2019_by_Glenn_Francis_%28cropped%29_3.jpg/800px-Taylor_Swift_2_-_2019_by_Glenn_Francis_%28cropped%29_3.jpg"
+wget "https://upload.wikimedia.org/wikipedia/commons/thumb/f/fd/El_Tres_de_Mayo%2C_by_Francisco_de_Goya%2C_from_Prado_thin_black_margin.jpg/1280px-El_Tres_de_Mayo%2C_by_Francisco_de_Goya%2C_from_Prado_thin_black_margin.jpg"
+wget "https://upload.wikimedia.org/wikipedia/commons/thumb/9/94/Robert_Downey_Jr_2014_Comic_Con_%28cropped%29.jpg/800px-Robert_Downey_Jr_2014_Comic_Con_%28cropped%29.jpg"
+
+echo "--- Download pretrained model"
+cd ../PAMA
+gdown "https://drive.google.com/uc?id=1rPB_qnelVVSad6CtadmhRFi0PMI_RKdy"
+
+mkdir -p checkpoints
+echo "--- Unpacking"
+unzip -j original_PAMA.zip -d checkpoints/
+echo "--- Done"

requirements.txt

@@ -0,0 +1,6 @@
+gdown
+numpy
+matplotlib
+Pillow
+torch>1.12
+torchvision