Update app.py

app.py

@@ -9,8 +9,7 @@ from HGTrainer import RobertaForValNovRegression, ValNovOutput
 
 @torch.no_grad()
 def classify(model, premise, conclusion):
-    model_path = "https://huggingface.co/spaces/pheinisch/" \
-                 "ConclusionValidityNoveltyClassifier-Augmentation/tree/main/models/"
+    model_path = "pheinisch/"
     if model == "model trained only with task-internal data":
         model_path += "task-internal-model--750-wo-synthetic"
     elif model == "model trained with 1k task-external+synthetic data":
@@ -18,7 +17,7 @@ def classify(model, premise, conclusion):
     elif model == "model trained with 10k task-internal-external-synthetic data":
         model_path += "task-internal-external-model--10k-synthetic"
     elif model == "model trained with 750 task-internal-synthetic data":
-        model_path += "synthetic-task-internal-model--751-CW-synthetic"
+        model_path += "ConclusionValidityNoveltyClassifier-Augmentation-in_syn_750"
 
     hg_model: RobertaForValNovRegression = RobertaForValNovRegression.from_pretrained(model_path)
     hg_model.loss = "ignore"