Update app.py

app.py

@@ -52,11 +52,11 @@ def CASPER(sue):
     allow=GPT2Tokenizer("承認").input_ids[1]
     deny=GPT2Tokenizer("否定").input_ids[1]
     probs=GPT2Model(**GPT2Tokenizer('CASPERは女としての人格を持っています。人間とCASPERの対話です。人間「'+sue+'。承認 か 否定 のどちらかで答えてください。」'+"CASPER 「",return_tensors="pt")).logits[0]
-    i=0
+    i=-1
     p_answer=probs
-    id=torch.argmax(probs[0])
-    votes.append(1 if probs[0][allow]>probs[0][deny] else -1)
-    return "承認" if probs[0][allow]>probs[0][deny] else "否定"
+    id=torch.argmax(probs[i])
+    votes.append(1 if probs[i][allow]>probs[i][deny] else -1)
+    return "承認" if probs[i][allow]>probs[i][deny] else "否定"
 
 def greet(sue):
     text1="BERT-1"+MELCHIOR(sue)