bashyaldhiraj2067 commited on
Commit
3a21e33
·
verified ·
1 Parent(s): 61170d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ class CharTokenizer:
41
  tokenizer = CharTokenizer(char2id, id2char)
42
 
43
  # Load T5 model
44
- # model_name = "bashyaldhiraj2067/t5_char_nepali"
45
- model_name = "bashyaldhiraj2067/attention_epoch_2_xpu_64_copymechanism_nepali_GEC_new_21"
46
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
47
 
48
  def correct_text(input_text, max_length=256):
 
41
  tokenizer = CharTokenizer(char2id, id2char)
42
 
43
  # Load T5 model
44
+ model_name = "bashyaldhiraj2067/t5_char_nepali"
45
+ # model_name = "bashyaldhiraj2067/attention_epoch_2_xpu_64_copymechanism_nepali_GEC_new_21"
46
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
47
 
48
  def correct_text(input_text, max_length=256):