Spaces:
Running
Running
Commit
·
79869da
1
Parent(s):
e9b6ed2
Flipped v2 to v1
Browse files- 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_v1
|
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 |
|