Update app.py

app.py

@@ -13,6 +13,11 @@ def sentiment(comment):
     model = pipeline("text-classification", model="JethroDD/twitter-roberta-base-sentiment-CustomModel_imdb")
     outputs = model(comment)
     result_text =outputs[0]["label"]
+    if result_text == "LABEL_1":
+        result_text = "Positive"
+    elif result_text == "LABEL_0":
+        result_text = "Negative"
+
     print(f"The predicted class is {result_text}")
     
     return result_text