Hugging Face's logo

Spaces:
prithivida
/
Gramformer
Running

App Files Community
1
prithivida commited on
Commit
403b4ed
•
1 Parent(s): 9886ca4

Update InferenceServer.py

Browse files
Files changed (1) hide show
  1. InferenceServer.py +1 -1
InferenceServer.py CHANGED
@@ -15,7 +15,7 @@ print("Loading models...")
15
  app = FastAPI()
16
 
17
  device = "cpu"
18
- correction_model_tag = "prithivida/grammar_error_correcter"
19
  correction_tokenizer = AutoTokenizer.from_pretrained(correction_model_tag)
20
  correction_model = AutoModelForSeq2SeqLM.from_pretrained(correction_model_tag)
21
 
15
  app = FastAPI()
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