taotao577 commited on
Commit
8eca82c
·
1 Parent(s): 394df7b

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ tokenizer1 = AutoTokenizer.from_pretrained(model1_checkpoint)
15
  model1 = AutoModelForSeq2SeqLM.from_pretrained(model1_checkpoint)
16
  model1 = model1.to(device)
17
 
18
- model1.load_state_dict(torch.load('../model/chineseToManju/epoch_41_valid_bleu_100.00_model_weights.bin'))
19
  model1.eval()
20
  # 2是满语到中文
21
  model2_checkpoint = "Helsinki-NLP/opus-mt-en-zh"
@@ -23,7 +23,7 @@ tokenizer2 = AutoTokenizer.from_pretrained(model2_checkpoint)
23
  model2 = AutoModelForSeq2SeqLM.from_pretrained(model2_checkpoint)
24
  model2 = model2.to(device)
25
 
26
- model2.load_state_dict(torch.load('../model/manjuToChinese/epoch_41_valid_bleu_0.00_model_weights.bin'))
27
  model2.eval()
28
 
29
  def chineseToManju(text):
 
15
  model1 = AutoModelForSeq2SeqLM.from_pretrained(model1_checkpoint)
16
  model1 = model1.to(device)
17
 
18
+ model1.load_state_dict(torch.load('epoch_41_valid_bleu_100.00_model_weights.bin'))
19
  model1.eval()
20
  # 2是满语到中文
21
  model2_checkpoint = "Helsinki-NLP/opus-mt-en-zh"
 
23
  model2 = AutoModelForSeq2SeqLM.from_pretrained(model2_checkpoint)
24
  model2 = model2.to(device)
25
 
26
+ model2.load_state_dict(torch.load('epoch_41_valid_bleu_0.00_model_weights.bin'))
27
  model2.eval()
28
 
29
  def chineseToManju(text):