Update app.py

app.py

@@ -22,9 +22,6 @@ configs = BaseModelConfigs.load("./configs.yaml")
 
 beam_settings = TTSettings(num_beams=5, min_length=1, max_length=100)
 
-import os
-os.environ['TRANSFORMERS_CACHE'] = 'cache/'
-
 app = FastAPI()
 
 origins = ["*"]
@@ -95,8 +92,8 @@ async def extract_text_from_image(image: UploadFile):
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 from pydantic import BaseModel
 
-tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large", cache_dir="cache/")
-chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large", cache_dir="cache/")
+tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large")
+chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
 
 class ChatPrompt(BaseModel):
     prompt: str