flexnotop commited on
Commit
350e1f5
1 Parent(s): 4318352

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,7 +9,10 @@ classifier=pipeline("text-classification",model="flexnotop/imdb-v1",device=0)
9
  if st.button("Classify"):
10
  if user_input:
11
  result=classifier(user_input)
12
- print(result);
 
 
 
13
  else:
14
  st.write("Please enter some text")
15
 
 
9
  if st.button("Classify"):
10
  if user_input:
11
  result=classifier(user_input)
12
+ if result[0]=="LABEL_1":
13
+ st.write("Positive Rating")
14
+ else:
15
+ st.write("Negative Rating")
16
  else:
17
  st.write("Please enter some text")
18