Spaces:
Running
Running
Update app.py
Browse files
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 |
|