Update app.py

app.py

@@ -21,7 +21,7 @@ def MELCHIOR(sue):
     #BERT
     allow=BERTTokenizer("承認").input_ids[1]
     deny=BERTTokenizer("否定").input_ids[1]
-    output=BERTModel(**BERTTokenizer('科学者としての人格を持ったMELCHIORは次の決議に答えます。人間「'+sue+'承認か否定どちらですか？」'+"MELCHIOR 「[MASK]」",return_tensors="pt")).logits
+    output=BERTModel(**BERTTokenizer('MELCHIORは科学者としての人格を持っています。人間とMELCHIORの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"MELCHIOR 「[MASK]」",return_tensors="pt")).logits
     BERTTokenizer.batch_decode(torch.argmax(output,-1))
     mask=output[0,-3,:]
     votes.append(1 if mask[allow]>mask[deny] else -1)
@@ -31,7 +31,7 @@ def BALTHASAR(sue):
     #mT5
     allow=mT5Tokenizer("承認").input_ids[1]
     deny=mT5Tokenizer("否定").input_ids[1]
-    encoder_output=mT5Model.encoder(**mT5Tokenizer('母としての人格を持ったBALTHASARは次の決議に答えます。人間「'+sue+'承認か否定どちらですか？」'+"BALTHASAR 「<X>」",return_tensors="pt"))
+    encoder_output=mT5Model.encoder(**mT5Tokenizer('BALTHASAR母としての人格としての人格を持っています。人間とBALTHASARの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"BALTHASAR 「<X>」",return_tensors="pt"))
     id=None
     p_answer=None
     probs=None
@@ -51,7 +51,7 @@ def CASPER(sue):
     #GPT2
     allow=GPT2Tokenizer("承認").input_ids[1]
     deny=GPT2Tokenizer("否定").input_ids[1]
-    probs=GPT2Model(**GPT2Tokenizer('女としての人格を持ったCASPERは次の決議に答えます。人間「'+sue+'承認か否定どちらですか？」'+"CASPER 「",return_tensors="pt")).logits[0]
+    probs=GPT2Model(**GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")).logits[0]
     i=0
     p_answer=probs
     id=torch.argmax(probs[0])