commit_5

app.py

@@ -47,7 +47,6 @@ with open(r'labels.txt', 'r') as fp:
 
 colormap = np.asarray(ade_palette())
 
-
 def label_to_color_image(label):
     if label.ndim != 2:
         raise ValueError("Expect 2-D input label")
@@ -56,7 +55,6 @@ def label_to_color_image(label):
         raise ValueError("label value too large.")
     return colormap[label]
 
-
 def draw_plot(pred_img, seg):
     fig = plt.figure(figsize=(20, 15))
 
@@ -78,7 +76,6 @@ def draw_plot(pred_img, seg):
     ax.tick_params(width=0.0, labelsize=25)
     return fig
 
-
 def sepia(input_img):
     input_img = Image.fromarray(input_img)
 
@@ -105,10 +102,9 @@ def sepia(input_img):
     fig = draw_plot(pred_img, seg)
     return fig
 
-
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(size=(400, 600)),
-                    outputs='plot',
+                    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')