Spaces:
Running
Running
Update models/hsd_tr.py
Browse files- models/hsd_tr.py +1 -1
models/hsd_tr.py
CHANGED
@@ -25,7 +25,7 @@ def write():
|
|
25 |
result_strength_tr = strength_pipeline_tr(tr_input)
|
26 |
strength_tr = result_strength_tr[0]["label"]
|
27 |
label_dict_strength = {'LABEL_0': 'Strength: 0', 'LABEL_1': 'Strength: 1', 'LABEL_2': 'Strength: 2','LABEL_3': 'Strength: 3', 'LABEL_4': 'Strength: 4'} #🚫
|
28 |
-
strength_tr = label_dict_strength[
|
29 |
st.success(sentiment_tr)
|
30 |
st.success(strength_tr)
|
31 |
|
|
|
25 |
result_strength_tr = strength_pipeline_tr(tr_input)
|
26 |
strength_tr = result_strength_tr[0]["label"]
|
27 |
label_dict_strength = {'LABEL_0': 'Strength: 0', 'LABEL_1': 'Strength: 1', 'LABEL_2': 'Strength: 2','LABEL_3': 'Strength: 3', 'LABEL_4': 'Strength: 4'} #🚫
|
28 |
+
strength_tr = label_dict_strength[strength_tr]
|
29 |
st.success(sentiment_tr)
|
30 |
st.success(strength_tr)
|
31 |
|