Rename app (1).py to app.py

app (1).py → app.py

@@ -8,10 +8,10 @@ import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "mattmdjaga/segformer_b2_clothes"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "mattmdjaga/segformer_b2_clothes"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 
 def ade_palette():
@@ -103,7 +103,7 @@ def sepia(input_img):
 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"],
+                    examples=["cityspace1.png", "cityspace2.png", ""cityspace3.png", "cityspace4.png", "cityspace5.png"],
                     allow_flagging='never')