Ankush05 commited on
Commit
2e02e9b
·
1 Parent(s): 166efca
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -3,6 +3,12 @@ from transformers import pipeline
3
 
4
 
5
  classifier = pipeline(model="distilbert-base-uncased-finetuned-sst-2-english")
 
 
 
 
 
 
6
  if classifier := st.chat_input("Classify: "):
7
  ans = classifier()
8
  with st.chat_message("User"):
 
3
 
4
 
5
  classifier = pipeline(model="distilbert-base-uncased-finetuned-sst-2-english")
6
+ classifier("This movie is disgustingly good !")
7
+ [{'label': 'POSITIVE', 'score': 1.0}]
8
+
9
+ classifier("Director tried too much.")
10
+ [{'label': 'NEGATIVE', 'score': 0.996}]
11
+
12
  if classifier := st.chat_input("Classify: "):
13
  ans = classifier()
14
  with st.chat_message("User"):