alexkueck commited on
Commit
2513b18
·
1 Parent(s): a413712

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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