.

app.py

@@ -18,30 +18,30 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [255, 0, 0],
-        [0, 255, 0],
-        [0, 0, 255],
-        [255, 255, 0],
-        [0, 255, 255],
-        [255, 0, 255],
-        [128, 0, 0],
-        [0, 128, 0],
-        [0, 0, 128],
-        [128, 128, 0],
-        [0, 128, 128],
-        [128, 0, 128],
-        [192, 192, 192],
-        [128, 128, 128],
-        [255, 128, 0],
-        [255, 0, 128],
-        [0, 255, 128],
-        [128, 255, 0]
+         [255, 0, 0],
+         [0, 255, 0],
+         [0, 0, 255],
+         [255, 255, 0],
+         [0, 255, 255],
+         [255, 0, 255],
+         [128, 0, 0],
+         [0, 128, 0],
+         [0, 0, 128],
+         [128, 128, 0],
+         [0, 128, 128],
+         [128, 0, 128],
+         [192, 192, 192],
+         [128, 128, 128],
+         [255, 128, 0],
+         [255, 0, 128],
+         [0, 255, 128],
+         [128, 255, 0]
     ]
 
 
 labels_list = []
 
-with open(r'labels.txt', 'r') as fp:
+with open(r"labels.txt", "r") as fp:
     for line in fp:
         labels_list.append(line[:-1])
 
@@ -64,7 +64,7 @@ def draw_plot(pred_img, seg):
 
     plt.subplot(grid_spec[0])
     plt.imshow(pred_img)
-    plt.axis('off')
+    plt.axis("off")
     LABEL_NAMES = np.asarray(labels_list)
     FULL_LABEL_MAP = np.arange(len(LABEL_NAMES)).reshape(len(LABEL_NAMES), 1)
     FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
@@ -106,10 +106,14 @@ def sepia(input_img):
     return fig
 
 
-demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(400, 600)),
-                    outputs=['plot'],
-                    examples=["person-1.jpg", "person-2.jpg","person-3.jpg","person-4.jpg","person-5.jpg"],
-                    allow_flagging='never')
+demo = gr.Interface(
+    fn=sepia,
+    inputs=gr.Image(shape=(400, 600)),
+    outputs=["plot"],
+    examples=[
+        "person-1.jpg","person-2.jpg","person-3.jpg","person-4.jpg", "person-5.jpg",],
+    allow_flagging="never",
+)
+
 
-demo.launch()
+demo.launch()