Mgolo commited on
Commit
45a8efc
·
verified ·
1 Parent(s): 604fa0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ whisper_model = None
27
  def load_darija_model():
28
  global translator
29
  if translator is None:
30
- model_name = "LocaleNLP/eng_wol"
31
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
32
  tokenizer = MarianTokenizer.from_pretrained(model_name)
33
  translator = pipeline("translation", model=model, tokenizer=tokenizer, device=0 if device.type == 'cuda' else -1)
 
27
  def load_darija_model():
28
  global translator
29
  if translator is None:
30
+ model_name = "LocaleNLP/eng_wolof"
31
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
32
  tokenizer = MarianTokenizer.from_pretrained(model_name)
33
  translator = pipeline("translation", model=model, tokenizer=tokenizer, device=0 if device.type == 'cuda' else -1)