miesnerjacob commited on
Commit
7b8b2a7
β€’
1 Parent(s): bc85a76

added to README

Browse files
Files changed (3) hide show
  1. README.md +8 -0
  2. app.py +3 -2
  3. emotion_detection.py +1 -1
README.md CHANGED
@@ -7,3 +7,11 @@ app_file: app.py
7
  python_version: 3.9
8
  pinned: false
9
  ---
 
 
 
 
 
 
 
 
 
7
  python_version: 3.9
8
  pinned: false
9
  ---
10
+
11
+ # Multi-task NLP
12
+
13
+ This application is a celebration of open-source and the power that programmers have been granted today by those who
14
+ give back to the community. This tool was constructed using Streamlit, Huggingface Transformers,
15
+ Transformers-Interpret, NLTK, Spacy, amongst other open-source Python libraries and models.
16
+
17
+ You can access a live version of this app here: https://huggingface.co/spaces/miesnerjacob/Multi-task-NLP
app.py CHANGED
@@ -96,8 +96,9 @@ if page == "Welcome!":
96
  * Emotion Detection
97
  * Named Entity Recognition
98
 
99
- More features may be added in the future, depending on community feedback. Please reach out to me at
100
- [email protected] or at my Linkedin page listed below if you have ideas or suggestions for improvement.
 
101
 
102
  If you would like to contribute yourself, feel free to fork the Github repository listed below and submit a merge request.
103
  """
 
96
  * Emotion Detection
97
  * Named Entity Recognition
98
 
99
+ More features may be added in the future including article/tweet/youtube input and model quality improvements,
100
+ depending on community feedback. Please reach out to me at [email protected] or at my Linkedin page listed
101
+ below if you have ideas or suggestions for improvement.
102
 
103
  If you would like to contribute yourself, feel free to fork the Github repository listed below and submit a merge request.
104
  """
emotion_detection.py CHANGED
@@ -5,7 +5,7 @@ import pandas as pd
5
 
6
 
7
  class EmotionDetection:
8
- def __init__(self, chunksize=512):
9
  hub_location = 'cardiffnlp/twitter-roberta-base-emotion'
10
  self.tokenizer = AutoTokenizer.from_pretrained(hub_location)
11
  self.model = AutoModelForSequenceClassification.from_pretrained(hub_location)
 
5
 
6
 
7
  class EmotionDetection:
8
+ def __init__(self):
9
  hub_location = 'cardiffnlp/twitter-roberta-base-emotion'
10
  self.tokenizer = AutoTokenizer.from_pretrained(hub_location)
11
  self.model = AutoModelForSequenceClassification.from_pretrained(hub_location)