Spaces:
Running
Running
Setup Gradio space
Browse files
app.py
CHANGED
@@ -7,10 +7,9 @@ CLASSES = {
|
|
7 |
'irrelevant': 1,
|
8 |
'no': 2,
|
9 |
}
|
10 |
-
|
11 |
-
model = AutoModelForSequenceClassification.from_pretrained(checkpoint_path)
|
12 |
model.eval()
|
13 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
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 |
|