Skip to content

Instantly share code, notes, and snippets.

from transformers import AutoModelForSeq2SeqLM
import torch
tag_1 = "prithivida/grammar_error_correcter"
tag_2 = "prithivida/grammar_error_correcter_v1"
correction_model_1 = AutoModelForSeq2SeqLM.from_pretrained(tag_1)
correction_model_2 = AutoModelForSeq2SeqLM.from_pretrained(tag_2)
params_1={}