richylyq commited on
Commit
b2d474f
1 Parent(s): bccf7be

try to reduce loading model time

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -26,6 +26,9 @@ import argparse
26
  import langid
27
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
28
 
 
 
 
29
 
30
  class myTheme(Base):
31
  def __init__(
@@ -128,8 +131,6 @@ def opus_trans(article, target_language):
128
  def nllb_trans(article, target_language):
129
  result_lang = detect_lang(article)
130
 
131
- tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
132
- model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
133
  inputs = tokenizer(article, return_tensors="pt")
134
 
135
  if target_language == "English":
 
26
  import langid
27
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
28
 
29
+ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
30
+ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
31
+
32
 
33
  class myTheme(Base):
34
  def __init__(
 
131
  def nllb_trans(article, target_language):
132
  result_lang = detect_lang(article)
133
 
 
 
134
  inputs = tokenizer(article, return_tensors="pt")
135
 
136
  if target_language == "English":