Update app.py

app.py

@@ -1,12 +1,12 @@
 # transformersとtorch、Gradioをインポート
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import numpy as np
 import gradio as gr
 
 # トークナイザーとモデルをロードする
 tokenizer = AutoTokenizer.from_pretrained("studio-ousia/luke-japanese-base-lite")
-model = AutoModel.from_pretrained("TomokiFujihara/luke-japanese-base-lite-offensiveness-estimation", trust_remote_code=True, token='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')
+model = AutoModelForSequenceClassification.from_pretrained("TomokiFujihara/luke-japanese-base-lite-offensiveness-estimation", trust_remote_code=True)
 
 # txt入力を受け取り、それを使用して新しいtxtを生成する関数
 def generate(text):