Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +2 -0
tasks/text.py
CHANGED
|
@@ -6,6 +6,8 @@ import random
|
|
| 6 |
import numpy as np
|
| 7 |
from huggingface_hub import PyTorchModelHubMixin
|
| 8 |
from tqdm import tqdm, trange
|
|
|
|
|
|
|
| 9 |
|
| 10 |
import torch
|
| 11 |
import torch.nn as nn
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
from huggingface_hub import PyTorchModelHubMixin
|
| 8 |
from tqdm import tqdm, trange
|
| 9 |
+
from sentence_transformers import SentenceTransformer
|
| 10 |
+
|
| 11 |
|
| 12 |
import torch
|
| 13 |
import torch.nn as nn
|