Spaces:
Sleeping
Sleeping
Update models/hsd_tr.py
Browse files- 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}")
|