Update tasks/text.py

tasks/text.py

@@ -16,10 +16,16 @@ DESCRIPTION = "electra fine tune"
 ROUTE = "/text"
 
 @router.post(ROUTE, tags=["Text Task"], 
-    """""
-    - Makes random predictions from the label space (0-7)
+             description=DESCRIPTION)
+
+async def evaluate_text(request: TextEvaluationRequest):
+    """
+    Evaluate text classification for climate disinformation detection.
+    
+    Current Model: Electra
     - Used as a baseline for comparison
     """
+    
     # Download from Google Drive
     import gdown
 
@@ -32,6 +38,32 @@ ROUTE = "/text"
     # Get space info
     username, space_url = get_space_info()
 
+    # Define the label mapping
+    LABEL_MAPPING = {
+        "0_not_relevant": 0,
+        "1_not_happening": 1,
+        "2_not_human": 2,
+        "3_not_bad": 3,
+        "4_solutions_harmful_unnecessary": 4,
+        "5_science_unreliable": 5,
+        "6_proponents_biased": 6,
+        "7_fossil_fuels_needed": 7
+    }
+
+    # Load and prepare the dataset
+    dataset = load_dataset(request.dataset_name)
+
+    # Convert string labels to integers
+    dataset = dataset.map(lambda x: {"label": LABEL_MAPPING[x["label"]]})
+
+    # Split dataset
+    train_test = dataset["train"]
+    test_dataset = dataset["test"]
+    
+    # Start tracking emissions
+    tracker.start()
+    tracker.start_task("inference")
+
     #--------------------------------------------------------------------------------------------   
     
     # Make random predictions (placeholder for actual model inference)
@@ -44,7 +76,6 @@ ROUTE = "/text"
     # Get true labels
     true_labels = test_dataset["label"]
 
-
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE
     #--------------------------------------------------------------------------------------------