prithivida commited on
Commit
f0e724c
1 Parent(s): 4e9016d

Update InferenceServer.py

Browse files
Files changed (1) hide show
  1. InferenceServer.py +2 -2
InferenceServer.py CHANGED
@@ -16,8 +16,8 @@ app = FastAPI()
16
 
17
  device = "cpu"
18
  correction_model_tag = "prithivida/grammar_error_correcter_v1"
19
- correction_tokenizer = AutoTokenizer.from_pretrained(correction_model_tag, use_auth_token=True)
20
- correction_model = AutoModelForSeq2SeqLM.from_pretrained(correction_model_tag, use_auth_token=True)
21
 
22
  def set_seed(seed):
23
  torch.manual_seed(seed)
 
16
 
17
  device = "cpu"
18
  correction_model_tag = "prithivida/grammar_error_correcter_v1"
19
+ correction_tokenizer = AutoTokenizer.from_pretrained(correction_model_tag)
20
+ correction_model = AutoModelForSeq2SeqLM.from_pretrained(correction_model_tag)
21
 
22
  def set_seed(seed):
23
  torch.manual_seed(seed)