update

README.md

@@ -12,6 +12,9 @@ Official implementation of DCT-Net for Full-body Portrait Stylization.
 > In: SIGGRAPH 2022 (**TOG**) 
 > *[arXiv preprint](https://arxiv.org/abs/2207.02426)* 
 
+<a href="https://colab.research.google.com/github/menyifang/DCT-Net/blob/main/notebooks/inference.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a> 
+[![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/SIGGRAPH2022/DCT-Net)
+
 
 ## Demo
 ![demo_vid](assets/demo.gif)
@@ -29,6 +32,7 @@ Official implementation of DCT-Net for Full-body Portrait Stylization.
 
 
 ## Web Demo
+- Integrated into [Colab notebook](https://colab.research.google.com/github/menyifang/DCT-Net/blob/main/notebooks/inference.ipynb). Try out the colab demo.
 - Integrated into [Huggingface Spaces 🤗](https://huggingface.co/spaces) using [Gradio](https://github.com/gradio-app/gradio). Try out the Web Demo [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/SIGGRAPH2022/DCT-Net)
 
 
@@ -41,6 +45,8 @@ Official implementation of DCT-Net for Full-body Portrait Stylization.
 
 
 ## Quick Start
+<a href="https://colab.research.google.com/github/menyifang/DCT-Net/blob/main/notebooks/inference.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="google colab logo"></a> 
+
 
 ```bash
 git clone https://github.com/menyifang/DCT-Net.git

notebooks/inference.ipynb

@@ -15,6 +15,15 @@
   }
  },
  "cells": [
+  {
+   "cell_type": "markdown",
+   "source": [
+    "x[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/menyifang/DCT-Net/blob/main/notebooks/inference.ipynb)"
+   ],
+   "metadata": {
+    "id": "D2MFmZtpVEp_"
+   }
+  },
   {
    "cell_type": "markdown",
    "source": [
@@ -199,12 +208,12 @@
     "vis_img('result_artstyle.png')"
    ],
    "metadata": {
-        "id": "k7Llm0SWMLKr",
-        "outputId": "20b665ce-653e-48a9-c489-d66e93e1907f",
     "colab": {
      "base_uri": "https://localhost:8080/",
      "height": 1000
-        }
+    },
+    "id": "k7Llm0SWMLKr",
+    "outputId": "20b665ce-653e-48a9-c489-d66e93e1907f"
    },
    "execution_count": 21,
    "outputs": [