Spaces:
Sleeping
Sleeping
Update models/hsd_ar.py
Browse files- models/hsd_ar.py +2 -3
models/hsd_ar.py
CHANGED
@@ -5,8 +5,7 @@ import time
|
|
5 |
from transformers import pipeline
|
6 |
import os
|
7 |
|
8 |
-
|
9 |
-
from transformers import pipeline
|
10 |
|
11 |
sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
|
12 |
|
@@ -21,4 +20,4 @@ if user_input:
|
|
21 |
sentiment = result[0]["label"]
|
22 |
|
23 |
sentiment = label_dict[sentiment]
|
24 |
-
st.write(f"
|
|
|
5 |
from transformers import pipeline
|
6 |
import os
|
7 |
|
8 |
+
|
|
|
9 |
|
10 |
sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
|
11 |
|
|
|
20 |
sentiment = result[0]["label"]
|
21 |
|
22 |
sentiment = label_dict[sentiment]
|
23 |
+
st.write(f"Detection: {sentiment}")
|