fixed enc-dec model

app.py

@@ -29,16 +29,13 @@ def BALTHASAR(sue):#mT5
     p_answer=None
     probs=None
     i=0
-    txt="<pad>"
-    probs=mT5Model(inputs_embeds=encoder_output.last_hidden_state,decoder_input_ids=mT5Tokenizer(txt,return_tensors="pt").input_ids).logits[0]
-    id=torch.argmax(probs[i+1])
-    txt=txt+"<X>"
+    txt="<pad><X>"
     i=i+1
-    probs=mT5Model(inputs_embeds=encoder_output.last_hidden_state,decoder_input_ids=mT5Tokenizer(txt,return_tensors="pt").input_ids).logits[0]
-    id=torch.argmax(probs[i+1])
+    probs=mT5Model(inputs_embeds=encoder_output.last_hidden_state,decoder_input_ids=mT5Tokenizer(txt,return_tensors="pt").input_ids[:,:-1]).logits[0]
+    id=torch.argmax(probs[-1])
     txt=txt+mT5Tokenizer.decode(id)
-    votes.append(1 if probs[i+1][allow]>probs[i+1][deny] else -1)
-    return "承認"  if probs[i+1][allow]>probs[i+1][deny] else "否定"
+    votes.append(1 if probs[-1][allow]>probs[-1][deny] else -1)
+    return "承認"  if probs[-1][allow]>probs[-1][deny] else "否定"
 
 def CASPER(sue):#GPT2
     allow=GPT2Tokenizer("承認").input_ids[1]