diff --git a/config.json b/config.json
index 06d8fa9..85f92b5 100644
--- a/config.json
+++ b/config.json
@@ -16,5 +16,6 @@
       "type": "image"
     }
   },
-  "bb_pretrained": false
+  "bb_pretrained": false,
+  "model_type": "birefnet"
 }
\ No newline at end of file