Hantr commited on
Commit
c421198
·
1 Parent(s): c8f1b79
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -5,8 +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, TFSegformerForSemanticSegmentation
9
- from transformers import BigBirdForImageCaptioning
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
  "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
 
5
  import numpy as np
6
  from PIL import Image
7
  import tensorflow as tf
8
+ from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation, BigBirdForImageCaptioning
 
9
 
10
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
11
  "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"