diff --git a/README.md b/README.md
index d201ccd..623f778 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@ python convert_weights.py
 
 **Inference**
 ```
-python test.py --input_dir [image_folder_path]
+python test.py --input_dir [image_folder_path] --device [cpu/cuda]
 ```