Update label.py

models/label.py

@@ -9,8 +9,8 @@ class PredictionRequest(BaseModel):
     model_config = ConfigDict(
         json_schema_extra={
             "example": {
-                "argument": "Climate change is accelerating due to industrial emissions.",
-                "key_point": "Human industry contributes significantly to global warming."
+                "argument": "School uniforms limit students' creativity",
+                "key_point": "Uniforms restrict personal freedom"
             }
         }
     )
@@ -31,11 +31,11 @@ class PredictionResponse(BaseModel):
         json_schema_extra={
             "example": {
                 "prediction": 1,
-                "confidence": 0.874,
+                "confidence": 0.956,
                 "label": "apparie",
                 "probabilities": {
-                    "non_apparie": 0.126,
-                    "apparie": 0.874
+                    "non_apparie": 0.044,
+                    "apparie": 0.956
                 }
             }
         }
@@ -57,24 +57,24 @@ class BatchPredictionRequest(BaseModel):
             "example": {
                 "pairs": [
                     {
-                        "argument": "School uniforms limit students' self-expression and creativity",
-                        "key_point": "Uniforms restrict personal freedom and individuality"
+                        "argument": "Students should wear school uniforms",
+                        "key_point": "School uniforms are good for students"
                     },
                     {
-                        "argument": "Renewable energy creates more jobs than fossil fuel industries",
-                        "key_point": "Green energy generates employment opportunities"
+                        "argument": "We need more renewable energy",
+                        "key_point": "Solar and wind power are important"
                     },
                     {
-                        "argument": "We should invest more in renewable energy to combat climate change",
-                        "key_point": "Capital punishment violates human rights"
+                        "argument": "Education should be free for everyone",
+                        "key_point": "Cars should be electric"
                     },
                     {
-                        "argument": "Online education provides flexibility and accessibility for all students",
-                        "key_point": "Digital learning offers convenient and inclusive education"
+                        "argument": "Exercise improves health",
+                        "key_point": "Physical activity is good for the body"
                     },
                     {
-                        "argument": "Vaccinations are essential for public health and disease prevention",
-                        "key_point": "Space exploration leads to scientific discoveries"
+                        "argument": "Reading books helps learning",
+                        "key_point": "We should build more parks"
                     }
                 ]
             }