Spaces:
Runtime error
Runtime error
Update prediction.py
Browse files- prediction.py +2 -2
prediction.py
CHANGED
@@ -155,9 +155,9 @@ def run():
|
|
155 |
y_pred_df = pd.DataFrame(y_pred_conv, columns=['0', '1', '2'])
|
156 |
y_pred_final=y_pred_df.idxmax(1).astype(int)
|
157 |
|
158 |
-
if y_pred_final
|
159 |
st.write('## Dude, your guest gave Positive feedback')
|
160 |
-
if y_pred_final
|
161 |
st.write('## Dude, your guest gave Neutral feedback')
|
162 |
else:
|
163 |
st.write('## Attention, your guest gave Negative feedback')
|
|
|
155 |
y_pred_df = pd.DataFrame(y_pred_conv, columns=['0', '1', '2'])
|
156 |
y_pred_final=y_pred_df.idxmax(1).astype(int)
|
157 |
|
158 |
+
if y_pred_final == 2:
|
159 |
st.write('## Dude, your guest gave Positive feedback')
|
160 |
+
if y_pred_final == 1:
|
161 |
st.write('## Dude, your guest gave Neutral feedback')
|
162 |
else:
|
163 |
st.write('## Attention, your guest gave Negative feedback')
|