Update app.py

app.py

@@ -60,7 +60,7 @@ def main():
             for entry in data:
                 label = entry["label"]
                 score = entry["score"]
-                print(f"Label: {label}, Score: {score}")
+                st.write(f"Label: {label}, Score: {score}")
 
         if analysis_type == "Named Entity Recognition":
             st.subheader("Named Entity Recognition:")
@@ -83,13 +83,12 @@ def main():
             
             # Parse JSON data
             data = toxicity_classifier(user_text)
-            st.write(data)
+            #st.write(data)
             # Extract and display label and score values
-            for labels_and_scores in data:
-                for entry in labels_and_scores:
-                    label = entry["label"]
-                    score = entry["score"]
-                    st.write(f"Label: {label}, Score: {score}")
+            for entry in data:
+                label = entry["label"]
+                score = entry["score"]
+                st.write(f"Label: {label}, Score: {score}")
 
 if __name__ == "__main__":
     main()