Update app.py
Browse files
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 += "
|
15 |
elif model == "model trained with 1k task-external+synthetic data":
|
16 |
-
model_path += "
|
17 |
elif model == "model trained with 10k task-internal-external-synthetic data":
|
18 |
-
model_path += "
|
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 |
|