added quotes

laurabarredaagusti · Oct 9, 2022 · cbc7d09 · cbc7d09
1 parent 035fc85
commit cbc7d09
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/app/app.py b/app/app.py
@@ -23,10 +23,10 @@ def predict():
 
     prediction = model.predict(df['text'])
     prediction = prediction[0]
-    # if prediction == 0:
-    #     prediction = 'The sentiment of this tweet is positive'
-    # else:
-    #     prediction = 'The sentiment of this tweet is negative'
+    if prediction == '0':
+        prediction = 'The sentiment of this tweet is positive'
+    else:
+        prediction = 'The sentiment of this tweet is negative'
     return render_template('predict.html', predict=prediction)
 
 if __name__ == '__main__':