Update app.py

app.py

@@ -9,9 +9,9 @@ import numpy as np
 import tensorflow as tf
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nvidia/segformer-b0-finetuned-ade-512-512")
+    "nvidia/segformer-b0-finetuned-ade-512-512", from_pt=True)
 model = SegformerForSemanticSegmentation.from_pretrained(
-    "segments-tobias/segformer-b0-finetuned-segments-sidewalk")
+    "segments-tobias/segformer-b0-finetuned-segments-sidewalk", from_pt=True)
 
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
@@ -117,7 +117,7 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(512, 512)),
+                    inputs=gr.Image(),
                     outputs=['plot'],
                     examples=["Sidewalk_1.jpg", "Sidewalk_2.jpg", "Sidewalk_3.jpg"],
                     allow_flagging='never')