update

README.md

@@ -1,6 +1,6 @@
 ---
 title: Controllable Multi-domain Semantic Artwork Synthesis
-emoji: 🚀
+emoji: 🖼️
 colorFrom: gray
 colorTo: pink
 sdk: docker

seg2art/inference_util.py

@@ -34,7 +34,7 @@ colors = np.array(colors)
 
 def remap_label(arr):
     # compare only last color channel to speed up
-    arr_b = arr[:, :, 0]
+    arr_b = arr[:, :, 2]
 
     # remap color to label
     for i in range(len(colors)):

static/index.js

@@ -52,7 +52,7 @@ cvsIn.addEventListener("mousemove", function (e) {
         ctxIn.lineTo(x, y);
         ctxIn.strokeStyle = ctxIn.color;
         ctxIn.stroke();
-        if (((new Date).getTime() - current_time) >= 400) {
+        if (((new Date).getTime() - current_time) >= 300) {
             move_range = domainSlider.value;
             onRecognition(move_range);
             current_time = (new Date).getTime();