Update app.py

app.py

@@ -13,7 +13,7 @@ def generate(text):
     # 入力テキストをトークナイズ
     inputs = tokenizer.encode_plus(text, return_tensors='pt')
     # トークナイズされたテキストを使用して攻撃性を推定
-    outputs = np.array(model(**inputs)[:3])
+    outputs = np.array(model(inputs['input_ids'], inputs['attention_mask'])[:3])
     # スコアを算出
     min = min(outputs)
     if min < 0: