gojiteji commited on
Commit
af48214
1 Parent(s): 108f368

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -25,9 +25,7 @@ def BALTHASAR(sue):#mT5
25
  allow=mBERTTokenizer("Yes").input_ids[1]
26
  deny=mBERTTokenizer("No").input_ids[1]
27
  output=mBERTModel(**mBERTTokenizer('BALTHASARは母としての人格を持っています。人間とBALTHASARの対話です。人間「'+sue+'。YesかNoか。」'+"BALTHASAR 「[MASK]」",return_tensors="pt")).logits
28
- print(mBERTTokenizer.batch_decode(torch.argmax(output,-1)))
29
  mask=output[0,-3,:]
30
- print(mBERTTokenizer.decode(torch.argmax(output[0,-3,:])))
31
  votes.append(1 if mask[allow]>mask[deny] else -1)
32
  return "承認" if mask[allow]>mask[deny] else "否定"
33
 
 
25
  allow=mBERTTokenizer("Yes").input_ids[1]
26
  deny=mBERTTokenizer("No").input_ids[1]
27
  output=mBERTModel(**mBERTTokenizer('BALTHASARは母としての人格を持っています。人間とBALTHASARの対話です。人間「'+sue+'。YesかNoか。」'+"BALTHASAR 「[MASK]」",return_tensors="pt")).logits
 
28
  mask=output[0,-3,:]
 
29
  votes.append(1 if mask[allow]>mask[deny] else -1)
30
  return "承認" if mask[allow]>mask[deny] else "否定"
31