egmaminta commited on
Commit
3703f24
·
1 Parent(s): 44f7165

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ from espnet2.bin.tts_inference import Text2Speech
8
  from espnet2.utils.types import str_or_none
9
  import call_labels
10
 
 
 
 
11
  # define the feature extractor
12
  ViT_extractor = AutoFeatureExtractor.from_pretrained('vincentclaes/mit-indoor-scenes')
13
 
 
8
  from espnet2.utils.types import str_or_none
9
  import call_labels
10
 
11
+ # call the labels (data set from: https://www.kaggle.com/itsahmad/indoor-scenes-cvpr-2019)
12
+ labels = call_labels.call_labels()
13
+
14
  # define the feature extractor
15
  ViT_extractor = AutoFeatureExtractor.from_pretrained('vincentclaes/mit-indoor-scenes')
16