Update app.py

app.py

@@ -10,21 +10,21 @@ text = st.text_area("Өгүүлбэр оруулна уу?")
 
 if text is not None:
     predictions = pipeline(text)
-
+    label = predictions[0]["label"]
+    probability = predictions[0]["score"]
     col1, col2 = st.columns(2)
     col1.header("Sentiment")
     col2.header("Probability")
 
-    for label, probability in zip(predictions[0]["label"], predictions[0]["score"]):
-        if label == "entailment":
-            sentiment = "Negative"
-        elif label == "contradiction":
-            sentiment = "Neutral"
-        elif label == "neutral":
-            sentiment = "Positive" 
+    if label == "entailment":
+        sentiment = "Negative"
+    elif label == "contradiction":
+        sentiment = "Neutral"
+    elif label == "neutral":
+        sentiment = "Positive" 
 
-        col1.write(sentiment)
-        col2.write(f"{probability:.2f}") 
+    col1.write(sentiment)
+    col2.write(f"{probability:.2f}") 
     # label = predictions[0]["label"]
     # probability = predictions[0]["score"]
     # col1.write(label)