Spaces:
Sleeping
Sleeping
Update models/hsd_tr.py
Browse files- models/hsd_tr.py +1 -2
models/hsd_tr.py
CHANGED
@@ -23,10 +23,9 @@ def write():
|
|
23 |
sentiment_tr = label_dict_sentiment[sentiment_tr]
|
24 |
|
25 |
result_strength_tr = strength_pipeline_tr(tr_input)
|
26 |
-
strength_tr =
|
27 |
label_dict_strength = {'LABEL_1': 'Strength: 1', 'LABEL_0': 'Strength: 0'} #🚫
|
28 |
strength_tr = label_dict_strength[strenght_tr]
|
29 |
-
|
30 |
st.success(sentiment_tr)
|
31 |
st.success(strength_tr)
|
32 |
|
|
|
23 |
sentiment_tr = label_dict_sentiment[sentiment_tr]
|
24 |
|
25 |
result_strength_tr = strength_pipeline_tr(tr_input)
|
26 |
+
strength_tr = result_strength_tr[0]["label"]
|
27 |
label_dict_strength = {'LABEL_1': 'Strength: 1', 'LABEL_0': 'Strength: 0'} #🚫
|
28 |
strength_tr = label_dict_strength[strenght_tr]
|
|
|
29 |
st.success(sentiment_tr)
|
30 |
st.success(strength_tr)
|
31 |
|