pheinisch commited on
Commit
4bdf884
·
1 Parent(s): 2d4559f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def classify(model, premise, conclusion):
20
  elif model == "model trained with 750 task-internal-synthetic data":
21
  model_path += "synthetic-task-internal-model--751-CW-synthetic"
22
 
23
- hg_model: RobertaForValNovRegression = RobertaForValNovRegression.from_pretrained(str(model.absolute()))
24
  hg_model.loss = "ignore"
25
  hg_tokenizer: transformers.RobertaTokenizer = \
26
  transformers.AutoTokenizer.from_pretrained("{}-base".format(hg_model.config.model_type))
 
20
  elif model == "model trained with 750 task-internal-synthetic data":
21
  model_path += "synthetic-task-internal-model--751-CW-synthetic"
22
 
23
+ hg_model: RobertaForValNovRegression = RobertaForValNovRegression.from_pretrained(model_path)
24
  hg_model.loss = "ignore"
25
  hg_tokenizer: transformers.RobertaTokenizer = \
26
  transformers.AutoTokenizer.from_pretrained("{}-base".format(hg_model.config.model_type))