th0mascat commited on
Commit
dc52995
1 Parent(s): aba7435

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ from happytransformer import HappyTextToText, TTSettings
20
  from transformers import AutoTokenizer, T5ForConditionalGeneration
21
  from pydantic import BaseModel
22
 
23
- tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large", cache_dir="./cache")
24
- chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large", cache_dir="./cache")
25
 
26
  configs = BaseModelConfigs.load("./configs.yaml")
27
 
 
20
  from transformers import AutoTokenizer, T5ForConditionalGeneration
21
  from pydantic import BaseModel
22
 
23
+ tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large")
24
+ chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
25
 
26
  configs = BaseModelConfigs.load("./configs.yaml")
27