Spaces:
Sleeping
Sleeping
Update models/hsd_ar.py
Browse files- models/hsd_ar.py +3 -5
models/hsd_ar.py
CHANGED
@@ -9,12 +9,10 @@ import os
|
|
9 |
|
10 |
sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
|
11 |
|
12 |
-
|
13 |
-
|
|
|
14 |
|
15 |
-
label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
|
16 |
-
|
17 |
-
user_input = st.text_input("")
|
18 |
if user_input:
|
19 |
result = sentiment_pipeline(user_input)
|
20 |
sentiment = result[0]["label"]
|
|
|
9 |
|
10 |
sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
|
11 |
|
12 |
+
user_input = st.text_input(
|
13 |
+
"Enter your text here:",
|
14 |
+
)
|
15 |
|
|
|
|
|
|
|
16 |
if user_input:
|
17 |
result = sentiment_pipeline(user_input)
|
18 |
sentiment = result[0]["label"]
|