aupfe08 commited on
Commit
3f387c5
Β·
verified Β·
1 Parent(s): 1ad1bc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
5
  import numpy as np
6
  from PIL import Image
7
  import tensorflow as tf
8
- from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
9
 
10
  feature_extractor = SegformerImageProcessor.from_pretrained(
11
  "nvidia/segformer-b5-finetuned-ade-640-640"
 
5
  import numpy as np
6
  from PIL import Image
7
  import tensorflow as tf
8
+ from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
9
 
10
  feature_extractor = SegformerImageProcessor.from_pretrained(
11
  "nvidia/segformer-b5-finetuned-ade-640-640"