gngpostalsrvc commited on
Commit
ad7f048
1 Parent(s): 3fe9b87

added application file

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,9 +5,11 @@ import spacy
5
  import librosa
6
  import gradio as gr
7
  import pandas as pd
8
- from transformers import pipeline
9
 
10
  asr = pipeline('automatic-speech-recognition', model='facebook/wav2vec2-large-960h-lv60-self')
 
 
11
  emo = pipeline('sentiment-analysis', model='arpanghoshal/EmoRoBERTa')
12
  lang_model = spacy.load("en_core_web_sm")
13
 
 
5
  import librosa
6
  import gradio as gr
7
  import pandas as pd
8
+ from transformers import pipeline, RobertaTokenizerFast, TFRobertaForSequenceClassification
9
 
10
  asr = pipeline('automatic-speech-recognition', model='facebook/wav2vec2-large-960h-lv60-self')
11
+ tokenizer = RobertaTokenizerFast.from_pretrained("arpanghoshal/EmoRoBERTa")
12
+ model = TFRobertaForSequenceClassification.from_pretrained("arpanghoshal/EmoRoBERTa")
13
  emo = pipeline('sentiment-analysis', model='arpanghoshal/EmoRoBERTa')
14
  lang_model = spacy.load("en_core_web_sm")
15