MENG21 commited on
Commit
b09a689
·
1 Parent(s): 7f80a20
Files changed (1) hide show
  1. HF_pipeline.py +2 -1
HF_pipeline.py CHANGED
@@ -16,4 +16,5 @@ def analyze_sintement(text, selected_model):
16
  classifier = pipeline("text-classification", model=MODEL_URLS[selected_model])
17
 
18
  result = classifier(text)
19
- return result
 
 
16
  classifier = pipeline("text-classification", model=MODEL_URLS[selected_model])
17
 
18
  result = classifier(text)
19
+ # st.text(result)
20
+ return result[0]['label'], result[0]['score']