Spaces:
Running
on
Zero
Running
on
Zero
osanseviero
commited on
Commit
•
86d577a
1
Parent(s):
74a629f
Update app.py
Browse files
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 |
|