VanshK04 commited on
Commit
2075500
·
verified ·
1 Parent(s): 5db411e

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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: