Spaces:

EUNSEO56
/

task3

Sleeping

EUNSEO56 commited on Nov 14, 2023

Commit

0dca934

•

1 Parent(s): 52d25cf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,8 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
 )

     "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
+    "nickmuchi/segformer-b4-finetuned-segments-sidewalk",
+    from_pt=True
 )