VanshK04 commited on
Commit
b361f9f
·
verified ·
1 Parent(s): e7811fb

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -2
tasks/text.py CHANGED
@@ -13,8 +13,8 @@ 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('bert_fine_tuned_model-20250107T090607Z-001.zip', 'r') as zip_ref:
18
  zip_ref.extractall(model_dir)
19
 
20
  # Load the fine-tuned BERT model and tokenizer
 
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:
18
  zip_ref.extractall(model_dir)
19
 
20
  # Load the fine-tuned BERT model and tokenizer