modify

app.py

@@ -10,10 +10,10 @@ import requests
 
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
+    "nvidia/segformer-b5-finetuned-ade-640-640"
 )
 model = SegformerForSemanticSegmentation.from_pretrained(
-    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
+    "nvidia/segformer-b5-finetuned-ade-640-640"
 )
 
 def ade_palette():
@@ -235,7 +235,7 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(640, 640)),
+                    inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     examples=["image-1.jpg", "image-2.jpg", "image-3.jpg", "image-4.jpeg", "image-5.jpg"],
                     allow_flagging='never')