Update tasks/text.py

tasks/text.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter
 from datetime import datetime
-from datasets import load_dataset
+from datasets import load_dataset, Dataset
 from sklearn.metrics import accuracy_score
 import random
 
@@ -9,6 +9,9 @@ from transformers import AutoTokenizer,BertForSequenceClassification,AutoModelFo
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
+import torch
+import numpy as np
+
 router = APIRouter()
 
 DESCRIPTION = "BERT V1.1"