Update app.py
Browse files
app.py
CHANGED
@@ -24,9 +24,9 @@ def main():
|
|
24 |
generated_text = keywords[0]['generated_text']
|
25 |
|
26 |
# Display the result based on sentiment
|
27 |
-
if sentiment_label == '
|
28 |
st.write(f"The hotel did bad in: {generated_text}")
|
29 |
-
else:
|
30 |
st.write(f"The hotel did good in: {generated_text}")
|
31 |
|
32 |
if __name__ == "__main__":
|
|
|
24 |
generated_text = keywords[0]['generated_text']
|
25 |
|
26 |
# Display the result based on sentiment
|
27 |
+
if sentiment_label == 'NEGATIVE':
|
28 |
st.write(f"The hotel did bad in: {generated_text}")
|
29 |
+
else:
|
30 |
st.write(f"The hotel did good in: {generated_text}")
|
31 |
|
32 |
if __name__ == "__main__":
|