pheinisch commited on
Commit
dfc5e48
1 Parent(s): 52e8cff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,11 +11,11 @@ from HGTrainer import RobertaForValNovRegression, ValNovOutput
11
  def classify(model, premise, conclusion):
12
  model_path = "pheinisch/"
13
  if model == "model trained only with task-internal data":
14
- model_path += "task-internal-model--750-wo-synthetic"
15
  elif model == "model trained with 1k task-external+synthetic data":
16
- model_path += "task-external-model--1k-synthetic"
17
  elif model == "model trained with 10k task-internal-external-synthetic data":
18
- model_path += "task-internal-external-model--10k-synthetic"
19
  elif model == "model trained with 750 task-internal-synthetic data":
20
  model_path += "ConclusionValidityNoveltyClassifier-Augmentation-in_syn_750"
21
 
 
11
  def classify(model, premise, conclusion):
12
  model_path = "pheinisch/"
13
  if model == "model trained only with task-internal data":
14
+ model_path += "ConclusionValidityNoveltyClassifier-Augmentation-in_750"
15
  elif model == "model trained with 1k task-external+synthetic data":
16
+ model_path += "ConclusionValidityNoveltyClassifier-Augmentation-ext_syn_1k"
17
  elif model == "model trained with 10k task-internal-external-synthetic data":
18
+ model_path += "ConclusionValidityNoveltyClassifier-Augmentation-int_ext_syn_10k"
19
  elif model == "model trained with 750 task-internal-synthetic data":
20
  model_path += "ConclusionValidityNoveltyClassifier-Augmentation-in_syn_750"
21