gojiteji commited on
Commit
3715289
1 Parent(s): df8a0e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -40,7 +40,8 @@ def BALTHASAR(sue):#mT5
40
  def CASPER(sue):#GPT2
41
  allow=GPT2Tokenizer("承認").input_ids[1]
42
  deny=GPT2Tokenizer("否定").input_ids[1]
43
- probs=GPT2Model(**GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")).logits[0]
 
44
  i=-1
45
  p_answer=probs
46
  id=torch.argmax(probs[i])
 
40
  def CASPER(sue):#GPT2
41
  allow=GPT2Tokenizer("承認").input_ids[1]
42
  deny=GPT2Tokenizer("否定").input_ids[1]
43
+ inpt=GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")
44
+ probs=GPT2Model(input_ids=inpt.input_ids[:,:-1],attention_mask=inpt.attention_mask[:,:-1]).logits[0]
45
  i=-1
46
  p_answer=probs
47
  id=torch.argmax(probs[i])