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],
|