Update main.py

main.py

@@ -4,16 +4,14 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 import re
 import os
-
+from transformers import AutoModelWithLMHead, AutoTokenizer
 
 app = FastAPI()
 
 
 
-from transformers import AutoModelWithLMHead, AutoTokenizer
-
-tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
-model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
+tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news", use_fast=False)
+model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-summarize-news", use_fast=False)
 
 def summarize(text, max_length=150):
   input_ids = tokenizer.encode(text, return_tensors="pt", add_special_tokens=True)