chopey commited on
Commit
641a0fe
1 Parent(s): 2e51a14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def transliteration(source_word):
9
  source_word_str = source_word.lower()
10
 
11
  tokenizer = T5Tokenizer.from_pretrained("chopey/dvt5-base")
12
- model = T5ForConditionalGeneration.from_pretrained('chopey/dvt5-base')
13
 
14
  input_ids = tokenizer.encode(source_word_str, return_tensors="pt")
15
  output_ids = model.generate(input_ids, max_length=512)
 
9
  source_word_str = source_word.lower()
10
 
11
  tokenizer = T5Tokenizer.from_pretrained("chopey/dvt5-base")
12
+ model = T5ForConditionalGeneration.from_pretrained('chopey/model_t5_base')
13
 
14
  input_ids = tokenizer.encode(source_word_str, return_tensors="pt")
15
  output_ids = model.generate(input_ids, max_length=512)