Update app.py

app.py

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