Update main.py

main.py

@@ -1,14 +1,14 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import T5ForConditionalGeneration, T5Tokenizer
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # 创建FastAPI实例
 app = FastAPI()
 
 # 加载T5模型和Tokenizer
-model_name = "google-t5/t5-small"
-model = T5ForConditionalGeneration.from_pretrained(model_name)
-tokenizer = T5Tokenizer.from_pretrained(model_name)
+model_name = "danibor/flan-t5-base-humanizer"
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # 定义输入数据的结构
 class ContentRequest(BaseModel):