gojiteji commited on
Commit
7ab5c6f
1 Parent(s): d969fcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,7 +21,7 @@ def MELCHIOR(sue):
21
  #BERT
22
  allow=BERTTokenizer("承認").input_ids[1]
23
  deny=BERTTokenizer("否定").input_ids[1]
24
- output=BERTModel(**BERTTokenizer('科学者としての人格を持ったMELCHIORは次の決議に答えます。人間「'+sue+'承認か否定どちらですか?」'+"MELCHIOR 「[MASK]」",return_tensors="pt")).logits
25
  BERTTokenizer.batch_decode(torch.argmax(output,-1))
26
  mask=output[0,-3,:]
27
  votes.append(1 if mask[allow]>mask[deny] else -1)
@@ -31,7 +31,7 @@ def BALTHASAR(sue):
31
  #mT5
32
  allow=mT5Tokenizer("承認").input_ids[1]
33
  deny=mT5Tokenizer("否定").input_ids[1]
34
- encoder_output=mT5Model.encoder(**mT5Tokenizer('母としての人格を持ったBALTHASARは次の決議に答えます。人間「'+sue+'承認か否定どちらですか?」'+"BALTHASAR 「<X>」",return_tensors="pt"))
35
  id=None
36
  p_answer=None
37
  probs=None
@@ -51,7 +51,7 @@ def CASPER(sue):
51
  #GPT2
52
  allow=GPT2Tokenizer("承認").input_ids[1]
53
  deny=GPT2Tokenizer("否定").input_ids[1]
54
- probs=GPT2Model(**GPT2Tokenizer('女としての人格を持ったCASPERは次の決議に答えます。人間「'+sue+'承認か否定どちらですか?」'+"CASPER 「",return_tensors="pt")).logits[0]
55
  i=0
56
  p_answer=probs
57
  id=torch.argmax(probs[0])
21
  #BERT
22
  allow=BERTTokenizer("承認").input_ids[1]
23
  deny=BERTTokenizer("否定").input_ids[1]
24
+ output=BERTModel(**BERTTokenizer('MELCHIORは科学者としての人格を持っています。人間とMELCHIORの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"MELCHIOR 「[MASK]」",return_tensors="pt")).logits
25
  BERTTokenizer.batch_decode(torch.argmax(output,-1))
26
  mask=output[0,-3,:]
27
  votes.append(1 if mask[allow]>mask[deny] else -1)
31
  #mT5
32
  allow=mT5Tokenizer("承認").input_ids[1]
33
  deny=mT5Tokenizer("否定").input_ids[1]
34
+ encoder_output=mT5Model.encoder(**mT5Tokenizer('BALTHASAR母としての人格としての人格を持っています。人間とBALTHASARの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"BALTHASAR 「<X>」",return_tensors="pt"))
35
  id=None
36
  p_answer=None
37
  probs=None
51
  #GPT2
52
  allow=GPT2Tokenizer("承認").input_ids[1]
53
  deny=GPT2Tokenizer("否定").input_ids[1]
54
+ probs=GPT2Model(**GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")).logits[0]
55
  i=0
56
  p_answer=probs
57
  id=torch.argmax(probs[0])