Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
4# Load Condition Checker
5classifier_path = "KomeijiForce/deberta-v3-large-check-scene"
6classifier_tokenizer = AutoTokenizer.from_pretrained(classifier_path)
7classifier = AutoModelForSequenceClassification.from_pretrained(classifier_path)
8
9# Formalize Condition Checking into Prompt
10scene = "Koishi quietly emerged from behind the dense foliage, her sudden appearance catching the corner of Satori's eye as she stepped into the direct line of vision, smiling mischievously."
11question = "Does Koishi enter someone's direct field of vision?"
12
13prompt = f'''Scene: {scene}
14
15Question: {question}
16
17Directly answer only yes/no/unknown.'''
18
19# Scoring and Decoding
20with torch.no_grad():
21 logits = classifier(**classifier_tokenizer(prompt, return_tensors="pt")).logits[0]
22 choice = logits.argmax(-1).item()
23
24answer = [False, None, True][choice]
25
26print(answer)
27# True