Application files

Procfile

@@ -0,0 +1 @@
+web: sh setup.sh && streamlit run app.py

app.py

@@ -0,0 +1,46 @@
+import time
+import streamlit as st
+from transformers import pipeline
+
+# Model to load
+MODEL_TO_LOAD = "swastik-kapture/offenseval-xlmr"
+TOKENIZER = "xlm-roberta-base"
+
+# create classification pipeline
+trained_model = pipeline("text-classification", model=MODEL_TO_LOAD, tokenizer=TOKENIZER)
+
+# Streamlit App
+def main():
+    # create a session state for conversation history
+    if 'conversation_history' not in st.session_state:
+        st.session_state.conversation_history = []
+    # streamlit title
+    st.title("OffensEval: Profanity Detection")
+    # user message
+    user_message = st.chat_input("Say something")
+    # if user input is present try to predict the outcome
+    if user_message:
+        # append user message to history
+        st.session_state.conversation_history.append(('user', user_message, time.time()))
+        # get predicted output
+        output = trained_model.predict(user_message)
+        # get predictied label and score
+        label = output[0]['label']
+        score = output[0]['score']
+        # default color
+        color = "white"
+        # get the color based on label
+        if label == "LABEL_0":
+            color = "green"
+            label = "No Offense"
+        elif label == "LABEL_1":
+            color = "red"
+            label = "Offensive"
+        st.session_state.conversation_history.append(('assistant', f"<div style='background-color: {color}; width: auto; height: 50px;'>Label: {label}; Score: {score:.2f}</div>", time.time()))
+    # Display chat history
+    for sender, message, timestamp in st.session_state.conversation_history:
+        with st.chat_message(sender):
+            st.write(message, unsafe_allow_html=True)
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,71 @@
+altair==5.1.2
+attrs==23.1.0
+blinker==1.6.3
+cachetools==5.3.1
+certifi==2023.7.22
+charset-normalizer==3.3.1
+click==8.1.7
+filelock==3.12.4
+fsspec==2023.10.0
+gitdb==4.0.11
+GitPython==3.1.40
+huggingface-hub==0.17.3
+idna==3.4
+importlib-metadata==6.8.0
+Jinja2==3.1.2
+jsonschema==4.19.1
+jsonschema-specifications==2023.7.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.2
+numpy==1.26.1
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.52
+nvidia-nvtx-cu12==12.1.105
+packaging==23.2
+pandas==2.1.1
+Pillow==10.1.0
+protobuf==4.24.4
+pyarrow==13.0.0
+pydeck==0.8.1b0
+Pygments==2.16.1
+python-dateutil==2.8.2
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.30.2
+regex==2023.10.3
+requests==2.31.0
+rich==13.6.0
+rpds-py==0.10.6
+safetensors==0.4.0
+six==1.16.0
+smmap==5.0.1
+streamlit==1.27.2
+sympy==1.12
+tenacity==8.2.3
+tokenizers==0.14.1
+toml==0.10.2
+toolz==0.12.0
+torch==2.1.0
+tornado==6.3.3
+tqdm==4.66.1
+transformers==4.34.1
+triton==2.1.0
+typing_extensions==4.8.0
+tzdata==2023.3
+tzlocal==5.2
+urllib3==2.0.7
+validators==0.22.0
+watchdog==3.0.0
+zipp==3.17.0

setup.sh

@@ -0,0 +1,8 @@
+mkdir -p ~/.streamlit/
+echo "\
+[server]\n\
+headless = true\n\
+port = $PORT\n\
+enableCORS = false\n\
+\n\
+" > ~/.streamlit/config.toml