Jeffrey Rathgeber Jr commited on
Commit
7749ef6
·
unverified ·
1 Parent(s): dbdfe6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import streamlit as st
 
2
  from transformers import pipeline
3
  # import numpy as np
4
 
@@ -13,4 +14,4 @@ st.write('You selected:', option)
13
  # pipeline
14
  preds = classifier(textIn)
15
  preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
16
- st.write('Input text is ', preds[0]['label'], ' with a confidence of ', preds[0]['score'])
 
1
  import streamlit as st
2
+ import tensorflow as tf
3
  from transformers import pipeline
4
  # import numpy as np
5
 
 
14
  # pipeline
15
  preds = classifier(textIn)
16
  preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
17
+ st.write('Input text is ', preds[0]['label'], ' with a confidence of ', preds[0]['score'])