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 |
|