kevinpro commited on
Commit
dca1b11
·
verified ·
1 Parent(s): 3de2dd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ flores_codes = list(code_mapping.keys())
30
  target_languages = [language for language in flores_codes if not language in REMOVED_TARGET_LANGUAGES]
31
 
32
  def load_model():
33
- model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
34
  print(f"Model loaded in {device}")
35
  return model
36
 
 
30
  target_languages = [language for language in flores_codes if not language in REMOVED_TARGET_LANGUAGES]
31
 
32
  def load_model():
33
+ model = AutoModel.from_pretrained(MODEL_NAME).to(device)
34
  print(f"Model loaded in {device}")
35
  return model
36