diff --git a/Cargo.toml b/Cargo.toml
index 0da9f5e..0b77857 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -4,7 +4,7 @@ version = "0.1.5"
 edition = "2024"
 repository = "https://github.com/jhqxxx/aha"
 license = "Apache-2.0"
-description = "aha model inference library, now supports Qwen2.5VL, MiniCPM4, VoxCPM, Qwen3VL, DeepSeek-OCR, Hunyuan-OCR, PaddleOCR-VL, VoxCPM1.5"
+description = "aha model inference library, now supports Qwen2.5VL, MiniCPM4, VoxCPM, Qwen3VL, DeepSeek-OCR, Hunyuan-OCR, PaddleOCR-VL, VoxCPM1.5, RMBG2.0"
 
 [dependencies]
 candle-core = { version = "0.9.1"}
diff --git a/rmbg_0.png b/rmbg_0.png
deleted file mode 100644
index c76d6bf..0000000
Binary files a/rmbg_0.png and /dev/null differ
diff --git a/tests/test_rmbg2_0.rs b/tests/test_rmbg2_0.rs
index 463c4ef..da4bb9b 100644
--- a/tests/test_rmbg2_0.rs
+++ b/tests/test_rmbg2_0.rs
@@ -37,11 +37,12 @@ fn rmbg2_0_generate() -> Result<()> {
 
     let i_start = Instant::now();
     let result = model.generate(mes)?;
+    let i_duration = i_start.elapsed();
+    println!("Time elapsed in generate is: {:?}", i_duration);
     for (i, img) in result.iter().enumerate() {
         let _ = img.save(format!("rmbg_{i}.png"));
     }
-    let i_duration = i_start.elapsed();
-    println!("Time elapsed in generate is: {:?}", i_duration);
+    
 
     Ok(())
 }