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 |
|