Spaces:

timmy0079
/

ML_test_2

Sleeping

timmy0079 commited on Nov 9, 2023

Commit

fc2beda

•

1 Parent(s): 8919890

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,9 +9,9 @@ import numpy as np
 import tensorflow as tf
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nvidia/segformer-b0-finetuned-ade-512-512", from_pt=True)
 model = SegformerForSemanticSegmentation.from_pretrained(
-    "segments-tobias/segformer-b0-finetuned-segments-sidewalk", from_pt=True)
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""

 import tensorflow as tf
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
+    "nvidia/segformer-b0-finetuned-ade-512-512")
 model = SegformerForSemanticSegmentation.from_pretrained(
+    "segments-tobias/segformer-b0-finetuned-segments-sidewalk")
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""