Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ from utils import *
|
|
10 |
from presets import *
|
11 |
from transformers import Trainer, TrainingArguments, DataCollatorForLanguageModeling
|
12 |
from setfit import SetFitModel, SetFitTrainer
|
|
|
13 |
import numpy as np
|
14 |
import evaluate
|
15 |
import pandas as pd
|
|
|
10 |
from presets import *
|
11 |
from transformers import Trainer, TrainingArguments, DataCollatorForLanguageModeling
|
12 |
from setfit import SetFitModel, SetFitTrainer
|
13 |
+
from sentence_transformers.losses import CosineSimilarityLoss
|
14 |
import numpy as np
|
15 |
import evaluate
|
16 |
import pandas as pd
|