token counting display only when tokens is passed

Author: mufeedvh

src/main.rs

@@ -176,14 +176,16 @@ fn main() -> Result<()> {
         println!("{}", serde_json::to_string_pretty(&json_output)?);
         return Ok(());
     } else {
-        println!(
-            "{}{}{} Token count: {}, Model info: {}",
-            "[".bold().white(),
-            "i".bold().blue(),
-            "]".bold().white(),
-            token_count.to_string().bold().yellow(),
-            model_info
-        );        
+        if args.tokens {
+            println!(
+                "{}{}{} Token count: {}, Model info: {}",
+                "[".bold().white(),
+                "i".bold().blue(),
+                "]".bold().white(),
+                token_count.to_string().bold().yellow(),
+                model_info
+            );
+        }
     }
 
     // Copy to Clipboard