add seg

app.py

@@ -44,7 +44,7 @@ for url in urls_mmpose:
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = Model_all(device)
 
-DESCRIPTION = '''# T2I-Adapter (Sketch & Keypose)
+DESCRIPTION = '''# T2I-Adapter (Sketch & Keypose & Segmentation)
 [Paper](https://arxiv.org/abs/2302.08453)               [GitHub](https://github.com/TencentARC/T2I-Adapter) 
 
 This gradio demo is for a simple experience of T2I-Adapter:

demo/model.py

@@ -287,7 +287,7 @@ class Model_all:
             labelmap = np.transpose(labelmap, (1,2,0))
             labelmap = cv2.resize(labelmap, (512, 512))
             labelmap = img2tensor(labelmap, bgr2rgb=False, float32=True)/255.
-            im_seg = tensor2img(labelmap)
+            im_seg = tensor2img(labelmap)[:,:,::-1]
             labelmap = labelmap.unsqueeze(0)
 
         # extract condition features