Spaces:

keras-io
/

Human-Part-Segmentation

Running on CPU Upgrade

satpalsr commited on Feb 10, 2022

Commit

04af290

•

1 Parent(s): e8261c9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 from huggingface_hub import from_pretrained_keras
 import cv2
-model = from_pretrained_keras("satpalsr/deeplabv3p-resnet50")
 colormap = np.array([[0,0,0], [31,119,180], [44,160,44], [44, 127, 125], [52, 225, 143],
                     [217, 222, 163], [254, 128, 37], [130, 162, 128], [121, 7, 166], [136, 183, 248],

 from huggingface_hub import from_pretrained_keras
 import cv2
+model = from_pretrained_keras("keras-io/deeplabv3p-resnet50")
 colormap = np.array([[0,0,0], [31,119,180], [44,160,44], [44, 127, 125], [52, 225, 143],
                     [217, 222, 163], [254, 128, 37], [130, 162, 128], [121, 7, 166], [136, 183, 248],