Cleanup

app.py

@@ -15,7 +15,7 @@ label_map = {0: "O", 1: "B-INDICATOR", 2: "I-INDICATOR", 3: "B-CAUSE", 4: "I-CAU
 
 # Streamlit App
 st.title("Attribution of Causality")
-st.write("Tags indicators and causes. GER only (for now)")
+st.write("Tags indicators and causes. GER only (atm)")
 
 # Text input for sentences
 sentences_input = st.text_area("Sentences (one per line)", "\n".join([
@@ -77,6 +77,6 @@ if st.button("Analyze Sentences"):
         output_sentence = " ".join(formatted_output)
 
         # Display formatted sentence with Streamlit
-        st.write(f"**Original Sentence:** {sentence}")
-        st.markdown(f"**Labeled Output:** {output_sentence}", unsafe_allow_html=True)
+        st.write(f"**Sentence:** {sentence}")
+        st.markdown(f"**Labeled:** {output_sentence}", unsafe_allow_html=True)
         st.write("---")