Spaces:

danu9327
/

333

Sleeping

redkye2 commited on Nov 14, 2023

Commit

241c7ea

1 Parent(s): 592a59a

2

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(
-    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024",
     from_pt=True
 )

 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
+    "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
+    "nvidia/segformer-b2-finetuned-cityscapes-1024-1024",
     from_pt=True
 )