arpm01 commited on
Commit
e1ff8b1
·
1 Parent(s): 553fe03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline, AutoFeatureExtractor, SegformerForSemanticSeg
5
  extractor = AutoFeatureExtractor.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
6
 
7
  model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
8
-
9
  pipe = pipeline(task="image-segmentation",
10
  model=model,
11
  feature_extractor=extractor)
 
5
  extractor = AutoFeatureExtractor.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
6
 
7
  model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
8
+ model.eval()
9
  pipe = pipeline(task="image-segmentation",
10
  model=model,
11
  feature_extractor=extractor)