osanseviero commited on
Commit
86d577a
1 Parent(s): 74a629f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -20,6 +20,7 @@ target_languages = [language for language in flores_codes if not language in REM
20
 
21
  def load_model():
22
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
 
23
  return model
24
 
25
 
 
20
 
21
  def load_model():
22
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
23
+ print(f"Model loaded in {device}")
24
  return model
25
 
26