gojiteji commited on
Commit
b286fa8
1 Parent(s): 2f45cbd

CASPERが未知語となる問題の修正

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -33,9 +33,9 @@ def BALTHASAR(sue):#mT5
33
 
34
 
35
  def CASPER(sue):#GPT2
36
- allow=GPT2Tokenizer("賛成").input_ids[1]
37
- deny=GPT2Tokenizer("反対").input_ids[1]
38
- inpt=GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。賛成か反対か。」'+"CASPER 「",return_tensors="pt")
39
  probs=GPT2Model(input_ids=inpt.input_ids[:,:-1],attention_mask=inpt.attention_mask[:,:-1]).logits[0]
40
  i=-1
41
  p_answer=probs
33
 
34
 
35
  def CASPER(sue):#GPT2
36
+ allow=GPT2Tokenizer("承認").input_ids[1]
37
+ deny=GPT2Tokenizer("否定").input_ids[1]
38
+ inpt=GPT2Tokenizer('女としての人格を持ったAI・カスパーと人間の対話です。人間「'+sue+'。これに承認か否定か。」'+"カスパー「私は,",return_tensors="pt")
39
  probs=GPT2Model(input_ids=inpt.input_ids[:,:-1],attention_mask=inpt.attention_mask[:,:-1]).logits[0]
40
  i=-1
41
  p_answer=probs