SoDehghan commited on
Commit
060c5f9
·
verified ·
1 Parent(s): 7fdd7f9

Update models/hsd_tr.py

Browse files
Files changed (1) hide show
  1. models/hsd_tr.py +1 -1
models/hsd_tr.py CHANGED
@@ -17,6 +17,6 @@ user_input = st.text_input("Enter your text here:",)
17
  if user_input:
18
  result = sentiment_pipeline(user_input)
19
  sentiment = result[0]["label"]
20
-
21
  sentiment = label_dict[sentiment]
22
  st.write(f"Detection: {sentiment}")
 
17
  if user_input:
18
  result = sentiment_pipeline(user_input)
19
  sentiment = result[0]["label"]
20
+ label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
21
  sentiment = label_dict[sentiment]
22
  st.write(f"Detection: {sentiment}")