Update to app

.gitignore

@@ -0,0 +1,2 @@
+venv
+.env

app.py

@@ -1,5 +1,7 @@
+import os
 import time
 import streamlit as st
+from dotenv import load_dotenv
 from transformers import pipeline
 
 # Model to load
@@ -7,7 +9,7 @@ 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, token="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")
+trained_model = pipeline("text-classification", model=MODEL_TO_LOAD, tokenizer=TOKENIZER, token=os.environ.get("HF_READ_KEY"))
 
 # Streamlit App
 def main():

requirements.txt

@@ -41,6 +41,7 @@ pyarrow==13.0.0
 pydeck==0.8.1b0
 Pygments==2.16.1
 python-dateutil==2.8.2
+python-dotenv==1.0.0
 pytz==2023.3.post1
 PyYAML==6.0.1
 referencing==0.30.2