Spaces:

prithivida
/

Gramformer

Running

prithivida commited on Oct 19, 2022

Commit

f0e724c

1 Parent(s): 4e9016d

Update InferenceServer.py

Files changed (1) hide show

InferenceServer.py CHANGED Viewed

@@ -16,8 +16,8 @@ app = FastAPI()
 device = "cpu"
 correction_model_tag = "prithivida/grammar_error_correcter_v1"
-correction_tokenizer = AutoTokenizer.from_pretrained(correction_model_tag, use_auth_token=True)
-correction_model     = AutoModelForSeq2SeqLM.from_pretrained(correction_model_tag, use_auth_token=True)
 def set_seed(seed):
     torch.manual_seed(seed)

 device = "cpu"
 correction_model_tag = "prithivida/grammar_error_correcter_v1"
+correction_tokenizer = AutoTokenizer.from_pretrained(correction_model_tag)
+correction_model     = AutoModelForSeq2SeqLM.from_pretrained(correction_model_tag)
 def set_seed(seed):
     torch.manual_seed(seed)