diff --git a/.ipynb_checkpoints/trytoubload-checkpoint.py b/.ipynb_checkpoints/trytoubload-checkpoint.py
index 23e61a5329316c52133e38eb1a49eaa365ca33a8..b4836269575d9e06fb159d8e9254c45927bc86e8 100644
--- a/.ipynb_checkpoints/trytoubload-checkpoint.py
+++ b/.ipynb_checkpoints/trytoubload-checkpoint.py
@@ -1,6 +1,18 @@
 from transformers import BertTokenizer, BertForSequenceClassification
 from peft import PeftModel
 import torch
+import os
+import pandas as pd
+import torch
+import numpy as np
+import matplotlib.pyplot as plt
+from sklearn.model_selection import train_test_split
+from sklearn.metrics import f1_score, precision_score, recall_score
+from sklearn.utils.class_weight import compute_class_weight
+from datasets import Dataset, load_from_disk
+from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer, TrainingArguments
+from torch import nn
+from peft import get_peft_model, LoraConfig, TaskType
 
 
 class MultiTaskBert(BertPreTrainedModel):
diff --git a/trytoubload.py b/trytoubload.py
index 23e61a5329316c52133e38eb1a49eaa365ca33a8..b4836269575d9e06fb159d8e9254c45927bc86e8 100644
--- a/trytoubload.py
+++ b/trytoubload.py
@@ -1,6 +1,18 @@
 from transformers import BertTokenizer, BertForSequenceClassification
 from peft import PeftModel
 import torch
+import os
+import pandas as pd
+import torch
+import numpy as np
+import matplotlib.pyplot as plt
+from sklearn.model_selection import train_test_split
+from sklearn.metrics import f1_score, precision_score, recall_score
+from sklearn.utils.class_weight import compute_class_weight
+from datasets import Dataset, load_from_disk
+from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer, TrainingArguments
+from torch import nn
+from peft import get_peft_model, LoraConfig, TaskType
 
 
 class MultiTaskBert(BertPreTrainedModel):