javear commited on
Commit
377b15c
1 Parent(s): a4ca8d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ encoded_input = tokenizer(text, return_tensors='pt')
8
  output = model(**encoded_input)
9
 
10
  results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5]
11
- return [ config.id2label[ids] for ids in results]
12
 
13
  #pipe = pipeline('sentiment-analysis')
14
  #text = st.text_area('enter some text!')
 
8
  output = model(**encoded_input)
9
 
10
  results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5]
11
+ #return [ config.id2label[ids] for ids in results]
12
 
13
  #pipe = pipeline('sentiment-analysis')
14
  #text = st.text_area('enter some text!')