Update app.py

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import pipeline
+from transformers import pipeline, DistilBertForSequenceClassification, AutoTokenizer
 import os
 import uvicorn
 
@@ -14,7 +14,9 @@ app = FastAPI()
 @app.on_event("startup")
 async def load_model():
     global sentiment_model
-    sentiment_model = pipeline("text-classification", model="tommyliphys/ai-detector-distilbert", from_tf=True)
+    model = DistilBertForSequenceClassification.from_pretrained("tommyliphys/ai-detector-distilbert", from_tf=True)
+    tokenizer = AutoTokenizer.from_pretrained("tommyliphys/ai-detector-distilbert")
+    sentiment_model = pipeline("text-classification", model=model, tokenizer=tokenizer)
 
 # 定义请求体的格式
 class TextRequest(BaseModel):