Merge pull request #4 from SudhanshuBlaze/SudhanshuDev

EDxHuggingface.py

@@ -10,6 +10,7 @@ load_dotenv()
 # AI model code
 HF_API_KEY = os.getenv("HF_API_KEY")
 
+# API_URL_ED = "https://api-inference.huggingface.co/models/j-hartmann/emotion-english-distilroberta-base" #alternate ED model(slow loading on first run)
 API_URL_ED = "https://api-inference.huggingface.co/models/bhadresh-savani/bert-base-go-emotion"
 API_URL_HS = "https://api-inference.huggingface.co/models/IMSyPP/hate_speech_en"
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
@@ -29,36 +30,37 @@ def query(payload):
 
 # Define color map for each emotion category
 color_map = {
-        'admiration': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'amusement': ['#ff7f0e', '#ffbb78', '#2ca02c', '#d62728'],
-        'anger': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'annoyance': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'approval': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'caring': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'confusion': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'curiosity': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'desire': ['#ff7f0e', '#ffbb78', '#2ca02c', '#d62728'],
-        'disappointment': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'disapproval': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'disgust': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'embarrassment': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'excitement': ['#ff7f0e', '#ffbb78', '#2ca02c', '#d62728'],
-        'fear': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'gratitude': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'grief': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'joy': ['#ff7f0e', '#ffbb78', '#2ca02c', '#d62728'],
-        'love': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'nervousness': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'optimism': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'pride': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'realization': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'relief': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728'],
-        'remorse': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'sadness': ['#d62728', '#ff9896', '#2ca02c', '#bcbd22'],
-        'surprise': ['#9467bd', '#c5b0d5', '#ff7f0e', '#d62728'],
-        'neutral': ['#1f77b4', '#aec7e8', '#ff7f0e', '#d62728']
+    'admiration': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'],
+    'amusement': ['#ff7f0e', '#98df8a', '#2ca02c', '#d62728'],
+    'anger': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'annoyance': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'approval': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'],
+    'caring': ['#98df8a', '#2ca02c', '#FF69B4', '#d62728'],
+    'confusion': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'],
+    'curiosity': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'],
+    'desire': ['#2ca02c', '#ff7f0e', '#98df8a', '#d62728'],
+    'disappointment': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'disapproval': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'disgust': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'embarrassment': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'],
+    'excitement': ['#ff7f0e', '#2ca02c', '#98df8a', '#d62728'],
+    'fear': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'gratitude': ['#98df8a', '#2ca02c', '#1f77b4', '#d62728'],
+    'grief': ['#ffbb78', '#d62728', '#bcbd22', '#ff7f0e'],
+    'joy': ['#ff7f0e', '#98df8a', '#2ca02c', '#d62728'],
+    'love': ['#FF69B4', '#98df8a', '#2ca02c', '#d62728'],
+    'nervousness': ['#ffbb78', '#ff7f0e', '#9467bd', '#d62728'],
+    'optimism': ['#98df8a', '#2ca02c', '#1f77b4', '#d62728'],
+    'pride': ['#98df8a', '#ff7f0e', '#1f77b4', '#d62728'],
+    'realization': ['#9467bd', '#ff7f0e', '#ffbb78', '#d62728'],
+    'relief': ['#1f77b4', '#98df8a', '#2ca02c', '#d62728'],
+    'remorse': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'sadness': ['#ffbb78', '#ff7f0e', '#d62728', '#bcbd22'],
+    'surprise': ['#ff7f0e', '#ffbb78', '#9467bd', '#d62728'],
+    'neutral': ['#2ca02c', '#98df8a', '#1f77b4', '#d62728']
 }
 
+
 # Labels for Hate Speech Classification
 label_hs = {"LABEL_0": "Acceptable", "LABEL_1": "inappropriate", "LABEL_2": "Offensive", "LABEL_3": "Violent"}
 
@@ -106,7 +108,7 @@ with st.sidebar:
 if submit:
 
     # Call API and get predicted probabilities for each emotion category and hate speech classification
-    payload = {"inputs": text_input, "use_cache": True, "wait_for_model": True}
+    payload = {"inputs": text_input, "options": {"wait_for_model": True, "use_cache": True}}
     response_ED, response_HS = query(payload)
     predicted_probabilities_ED = response_ED[0]
     predicted_probabilities_HS = response_HS[0]