Spaces:
Running
Running
jhj0517
commited on
Commit
·
99a813f
1
Parent(s):
a3c5455
Update print
Browse files
tests/test_translation.py
CHANGED
@@ -15,7 +15,7 @@ def test_nllb_inference(
|
|
15 |
file_path: str
|
16 |
):
|
17 |
nllb_inferencer = NLLBInference()
|
18 |
-
print("Device :
|
19 |
|
20 |
result_str, file_paths = nllb_inferencer.translate_file(
|
21 |
fileobjs=[file_path],
|
|
|
15 |
file_path: str
|
16 |
):
|
17 |
nllb_inferencer = NLLBInference()
|
18 |
+
print(f"NLLB Device : {nllb_inferencer.device}")
|
19 |
|
20 |
result_str, file_paths = nllb_inferencer.translate_file(
|
21 |
fileobjs=[file_path],
|