Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +9 -0
tasks/text.py
CHANGED
@@ -2,7 +2,16 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
2 |
import torch
|
3 |
import os
|
4 |
import zipfile
|
|
|
|
|
|
|
|
|
|
|
5 |
|
|
|
|
|
|
|
|
|
6 |
# Unzip the uploaded file
|
7 |
model_dir = "/tasks/bert_fine_tuned_model"
|
8 |
with zipfile.ZipFile('/tasks/bert_fine_tuned_model-20250107T090607Z-001.zip', 'r') as zip_ref:
|
|
|
2 |
import torch
|
3 |
import os
|
4 |
import zipfile
|
5 |
+
from fastapi import APIRouter
|
6 |
+
from datetime import datetime
|
7 |
+
from datasets import load_dataset
|
8 |
+
from sklearn.metrics import accuracy_score
|
9 |
+
import random
|
10 |
|
11 |
+
from .utils.evaluation import TextEvaluationRequest
|
12 |
+
from .utils.emissions import tracker, clean_emissions_data, get_space_info
|
13 |
+
|
14 |
+
router = APIRouter()
|
15 |
# Unzip the uploaded file
|
16 |
model_dir = "/tasks/bert_fine_tuned_model"
|
17 |
with zipfile.ZipFile('/tasks/bert_fine_tuned_model-20250107T090607Z-001.zip', 'r') as zip_ref:
|