Update app.py

app.py

@@ -23,7 +23,7 @@ configs = BaseModelConfigs.load("./configs.yaml")
 beam_settings = TTSettings(num_beams=5, min_length=1, max_length=100)
 
 import os
-os.environ['TRANSFORMERS_CACHE'] = './cache/'
+os.environ['TRANSFORMERS_CACHE'] = 'cache/'
 
 app = FastAPI()
 
@@ -95,8 +95,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")
-chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
+tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large", cache_dir="cache/")
+chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large", cache_dir="cache/")
 
 class ChatPrompt(BaseModel):
     prompt: str