MrPio commited on
Commit
8b8a073
·
1 Parent(s): a7a36a7

Setup Gradio space

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,10 +7,9 @@ CLASSES = {
7
  'irrelevant': 1,
8
  'no': 2,
9
  }
10
- checkpoint_path = "MrPio/TheSeagullStory-nli-deberta-v3-base"
11
- model = AutoModelForSequenceClassification.from_pretrained(checkpoint_path)
12
  model.eval()
13
- tokenizer = AutoTokenizer.from_pretrained(checkpoint_path)
14
  story = open('story.txt').read().replace("\n\n", "\n").replace("\n", " ").strip()
15
 
16
 
 
7
  'irrelevant': 1,
8
  'no': 2,
9
  }
10
+ model = AutoModelForSequenceClassification.from_pretrained('MrPio/TheSeagullStory-nli-deberta-v3-base')
 
11
  model.eval()
12
+ tokenizer = AutoTokenizer.from_pretrained('cross-encoder/nli-deberta-v3-base')
13
  story = open('story.txt').read().replace("\n\n", "\n").replace("\n", " ").strip()
14
 
15