input details corrected

extractor.py

@@ -15,8 +15,8 @@ def get_card_xy(model_path, image_path):
     st.write(f"{input_details}")
 
     # Obtain the height and width of the corresponding image from the input tensor
-    image_height = input_details[0]['shape'][1] # 640
-    image_width = input_details[0]['shape'][2] # 640
+    image_height = input_details[0]['shape'][2] # 640
+    image_width = input_details[0]['shape'][3] # 640
 
     # Image Preparation
     # image_name = 'car.jpg'
@@ -28,7 +28,7 @@ def get_card_xy(model_path, image_path):
     image_np = image_np[np.newaxis, :]
 
     # inference
-    interpreter.set_tensor(input_details[0]['index'], image_np)
+    interpreter.set_tensor(input_details[0]['index']+1, image_np)
     interpreter.invoke()
 
     # Obtaining output results