Hugging Face's logo

Spaces:
timmy0079
/
ML_test_2
Sleeping

App Files Community
timmy0079 commited on
Commit
5d735d8
•
1 Parent(s): e4c0dca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,9 +9,11 @@ import numpy as np
9
  import tensorflow as tf
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
- "nvidia/segformer-b0-finetuned-ade-512-512")
 
13
  model = SegformerForSemanticSegmentation.from_pretrained(
14
- "segments-tobias/segformer-b0-finetuned-segments-sidewalk")
 
15
 
16
  def ade_palette():
17
  """ADE20K palette that maps each class to RGB values."""
 
9
  import tensorflow as tf
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
+ "nielsr/segformer-b0-finetuned-segments-sidewalk",
13
+ from_pt=True)
14
  model = SegformerForSemanticSegmentation.from_pretrained(
15
+ "nielsr/segformer-b0-finetuned-segments-sidewalk",
16
+ from_pt=True)
17
 
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""