Update HebEMO.py

HebEMO.py

@@ -1,5 +1,5 @@
 class HebEMO:
-    def __init__(self, device=0, emotions = ['expectation', 'happy', 'trust', 'fear', 'surprise', 'anger',
+    def __init__(self, device=0, emotions = ['anticipation', 'joy', 'trust', 'fear', 'surprise', 'anger',
       'sadness', 'disgust']):
 
         from transformers import pipeline
@@ -12,8 +12,8 @@ class HebEMO:
         for emo in tqdm(emotions): 
             self.hebemo_models[emo] = pipeline(
                 "sentiment-analysis",
-                model="../hebEMO/"+emo+'_classifier',
-                tokenizer="../heBERT_base_oscar",
+                model="avichr/hebEMO_"+emo,
+                tokenizer="avichr/heBERT",
                 device = self.device #run on GPU
             )
     
@@ -73,7 +73,6 @@ class HebEMO:
             hebEMO = pd.DataFrame()
             for emo in hebEMO_df.columns[1::2]:
                 hebEMO[emo] = abs(hebEMO_df[emo]-(1-hebEMO_df['confidence_'+emo]))
-                hebEMO = hebEMO.rename(columns= {'happy': 'joy', 'expectation':'anticipation'})
 
             for i in range(0,1):    
                 ax = plutchik(hebEMO.to_dict(orient='records')[i])