[edit] update readme

README.md

@@ -35,21 +35,21 @@ The model is trained to predict in the following schema
 from enum import Enum
 from pydantic import BaseModel
 
-class RelationshipStatus(str, Enum):
+class EmotionLevel(str, Enum):
     na = "na"
     low = "low"
     medium = "medium"
     high = "high"
     
 class EmotionLabel(BaseModel):
-    joy: RelationshipStatus
-    trust: RelationshipStatus
-    fear: RelationshipStatus
-    surprise: RelationshipStatus
-    sadness: RelationshipStatus
-    disgust: RelationshipStatus
-    anger: RelationshipStatus
-    anticipation: RelationshipStatus
+    joy: EmotionLevel
+    trust: EmotionLevel
+    fear: EmotionLevel
+    surprise: EmotionLevel
+    sadness: EmotionLevel
+    disgust: EmotionLevel
+    anger: EmotionLevel
+    anticipation: EmotionLevel
         
 class EntryResult(BaseModel):
     emotion: EmotionLabel
@@ -81,7 +81,7 @@ input_text = tokenizer.apply_chat_template(
     add_generation_prompt=True,
 )
 prediction = generator(input_text)
->>> EntryResult(emotion=EmotionLabel(joy=<RelationshipStatus.na: 'na'>, ...)
+>>> EntryResult(emotion=EmotionLabel(joy=<EmotionLevel.na: 'na'>, ...)
 ```
 
 Using endpoint loaded with vllm & OpenAI client package