Spaces:

th0mascat
/

test

Running

th0mascat commited on Feb 5

Commit

dc52995

•

1 Parent(s): aba7435

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,8 +20,8 @@ from happytransformer import HappyTextToText, TTSettings
 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")
 configs = BaseModelConfigs.load("./configs.yaml")

 from transformers import AutoTokenizer, T5ForConditionalGeneration
 from pydantic import BaseModel
+tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-large")
+chatModel = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-large")
 configs = BaseModelConfigs.load("./configs.yaml")