update predict verbose (#514)

obss · Jun 28, 2022 · 70b039f · 70b039f
1 parent 5e45d75
commit 70b039f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sahi/predict.py b/sahi/predict.py
@@ -520,7 +520,10 @@ def predict(
 
         durations_in_seconds["prediction"] += prediction_result.durations_in_seconds["prediction"]
         # Show prediction time
-        tqdm.write("Prediction time is: {:.2f} ms".format(prediction_result.durations_in_seconds["prediction"] * 1000))
+        if verbose:
+            tqdm.write(
+                "Prediction time is: {:.2f} ms".format(prediction_result.durations_in_seconds["prediction"] * 1000)
+            )
 
         if dataset_json_path:
             if source_is_video is True: