From fccbaa7b2d0357e97a699459868e0ef3329f60a1 Mon Sep 17 00:00:00 2001
From: Devraj Kashyap <96662811+dev-rajk@users.noreply.github.com>
Date: Mon, 16 Dec 2024 01:38:51 +0530
Subject: [PATCH] Update README.md

---
 README.md | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 2d91dc0..fa0d7f1 100644
--- a/README.md
+++ b/README.md
@@ -37,11 +37,9 @@ from YOLOv8_Explainer import yolov8_heatmap, display_images
 
 model = yolov8_heatmap(
     weight="/location/model.pt", 
-        conf_threshold=0.4, 
-        device = "cpu", 
+        conf_threshold=0.4,  
         method = "EigenCAM", 
         layer=[10, 12, 14, 16, 18, -3],
-        backward_type="all",
         ratio=0.02,
         show_box=True,
         renormalize=False,