Spaces:

tu11p
/

gradio_seg_sidewalk

Runtime error

tu111p commited on Nov 9, 2023

Commit

6fd06e4

•

1 Parent(s): 9129006

change model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,10 +8,10 @@ import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nickmuchi/segformer-b4-finetuned-segments-sidewalk", from_pt=True
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "nickmuchi/segformer-b4-finetuned-segments-sidewalk", from_pt=True
 )
 def ade_palette():

 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
+    "segments-tobias/segformer-b3-finetuned-segments-sidewalk", from_pt=True
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
+    "segments-tobias/segformer-b3-finetuned-segments-sidewalk", from_pt=True
 )
 def ade_palette():