Update tasks/audio.py

tasks/audio.py

@@ -263,31 +263,29 @@ true_labels = test_dataset["label"]
 accuracy = accuracy_score(true_labels, numeric_predictions)
 print("Accuracy:", accuracy)
 
-    #--------------------------------------------------------------------------------------------
-    # YOUR MODEL INFERENCE STOPS HERE
-    #--------------------------------------------------------------------------------------------  
-   
-    # Stop tracking emissions
-    emissions_data = tracker.stop_task()
-   
-   
-   
-    # Prepare results dictionary
-    results = {
-        "username": username,
-        "space_url": space_url,
-        "submission_timestamp": datetime.now().isoformat(),
-        "model_description": DESCRIPTION,
-        "accuracy": float(accuracy),
-        "energy_consumed_wh": emissions_data.energy_consumed * 1000,
-        "emissions_gco2eq": emissions_data.emissions * 1000,
-        "emissions_data": clean_emissions_data(emissions_data),
-        "api_route": ROUTE,
-        "dataset_config": {
-            "dataset_name": request.dataset_name,
-            "test_size": request.test_size,
-            "test_seed": request.test_seed
-        }
+#--------------------------------------------------------------------------------------------
+# YOUR MODEL INFERENCE STOPS HERE
+#--------------------------------------------------------------------------------------------  
+
+# Stop tracking emissions
+emissions_data = tracker.stop_task()
+
+# Prepare results dictionary
+results = {
+    "username": username,
+    "space_url": space_url,
+    "submission_timestamp": datetime.now().isoformat(),
+    "model_description": DESCRIPTION,
+    "accuracy": float(accuracy),
+    "energy_consumed_wh": emissions_data.energy_consumed * 1000,
+    "emissions_gco2eq": emissions_data.emissions * 1000,
+    "emissions_data": clean_emissions_data(emissions_data),
+    "api_route": ROUTE,
+    "dataset_config": {
+        "dataset_name": request.dataset_name,
+        "test_size": request.test_size,
+        "test_seed": request.test_seed
     }
-   
-    return results 
+}
+
+return results