Update app.py

app.py

@@ -1,42 +1,77 @@
-import textwrap
-import subprocess
-import shutil
 import os
+import shutil
+import subprocess
+import textwrap
 from pathlib import Path
 
-import torch
 import gradio as gr
+import torch
 from huggingface_hub import hf_hub_download
 
-
 REPO_ID = "kbrodt/sketch2pose"
 API_TOKEN = os.environ["sketch2pose"]
 ASSET_DIR = Path("./assets")
-
-filename = "models_smplx_v1_1.zip"
-smpl_path = hf_hub_download(
-    repo_id=REPO_ID,
-    repo_type="model",
-    filename=filename,
-    use_auth_token=API_TOKEN,
-    cache_dir=ASSET_DIR,
-)
-if not (ASSET_DIR / filename).is_file():
-    shutil.copy(smpl_path, ASSET_DIR)
-
-subprocess.run("bash ./scripts/download.sh".split())
-subprocess.run("bash ./scripts/prepare.sh".split())
-
-
 SAVE_DIR = "output"
 CMD = textwrap.dedent("""
     python src/pose.py
         --save-path {}
         --img-path {}
 """)
+TITLE = "Sketch2Pose: Estimating a 3D Character Pose from a Bitmap Sketch"
+DESCRIPTION = '''
+<table>
+<th>
+<ul>
+<li><strong>Project</strong> <a href="http://www-labs.iro.umontreal.ca/~bmpix/sketch2pose/">sketch2pose</a></li>
+<li><strong>Code</strong> <a href="https://github.com/kbrodt/sketch2pose">kbrodt/sketch2pose</a>
+<li><strong>Paper</strong> <a href="https://dl.acm.org/doi/10.1145/3528223.3530106">ACM</a>
+</ul>
+<iframe src="https://ghbtns.com/github-btn.html?user=kbrodt&repo=sketch2pose&type=star&count=true&v=2&size=small" frameborder="0" scrolling="0" width="100" height="20"></iframe>
+</th>
+<th>
+<video controls autoplay muted loop>
+<source src="http://www-labs.iro.umontreal.ca/~bmpix/sketch2pose/sketch2pose.webm" type="video/mp4">
+</video>
+</th>
+</table>
+
+<details>
+<summary>Citation</summary>
+```
+@article{brodt2022sketch2pose,
+    author = {Kirill Brodt and Mikhail Bessmeltsev},
+    title = {Sketch2Pose: Estimating a 3D Character Pose from a Bitmap Sketch},
+    journal = {ACM Transactions on Graphics},
+    year = {2022},
+    month = {7},
+    volume = {41},
+    number = {4},
+    doi = {10.1145/3528223.3530106},
+}
+```
+</details>
+'''
+
+
+def prepare():
+    filename = "models_smplx_v1_1.zip"
+    smpl_path = hf_hub_download(
+        repo_id=REPO_ID,
+        repo_type="model",
+        filename=filename,
+        use_auth_token=API_TOKEN,
+        cache_dir=ASSET_DIR,
+    )
+    if not (ASSET_DIR / filename).is_file():
+        shutil.copy(smpl_path, ASSET_DIR)
+    
+    subprocess.run("bash ./scripts/download.sh".split())
+    subprocess.run("bash ./scripts/prepare.sh".split())
 
 
 def main():
+    prepare()
+
     save_dir = Path(SAVE_DIR)
     save_dir.mkdir(parents=True, exist_ok=True)
 
@@ -77,7 +112,9 @@ def main():
             gr.Checkbox(value=True, label="Pose naturalness"),
         ],
         outputs=gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="SMPL 3D pose"),
-        examples=examples,
+        examples=examples,        
+        title=TITLE,
+        description=DESCRIPTION,
     )
 
     demo.launch()