Update app.py

app.py

@@ -44,12 +44,14 @@ if generate_button:
         # Call the model with the input text
         input_text = "I love you"
         model_output = classifier(input_text)
-
-        length = len(model_output)
-        model_output_first = model_output[0]
+        output = model_output[0]
         
         # Displaying the input and model's output
-        st.write(f"Input Text: {input_text}")
-        st.write("Model Output:", model_output)
-        st.write("Model Output 2:", model_output_first)
-        st.write("Length:", length)
+        st.write(f"Generated Tweet: {input_text}")
+        for i in range(3):
+            if output[i]['label'] == 'LABEL_0':
+                st.write(f"Non-Toxic Content: {output[i]['score']:.2f}%")
+            elif output[i]['label'] == 'LABEL_1':
+                st.write(f"Toxic Content: {output[i]['score']:.2f}%")
+            else:
+                continue