From 97ec41f9a73c1cc7171a2901bae110b353ad7431 Mon Sep 17 00:00:00 2001
From: Quaggles <8382945+Quaggles@users.noreply.github.com>
Date: Wed, 22 Jan 2025 16:45:58 +1000
Subject: [PATCH] Fixes FileNotFoundError error with nsfw detector

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index baea838..b4e65bf 100644
--- a/nodes.py
+++ b/nodes.py
@@ -82,7 +82,7 @@ if not os.path.exists(NSFWDET_MODEL_PATH):
     ]
     for model_url in nd_urls:
         model_name = os.path.basename(model_url)
-        model_path = os.path.join("vit-base-nsfw-detector", model_name)
+        model_path = os.path.join(NSFWDET_MODEL_PATH, model_name)
         download(model_url, model_path, model_name)
 
 dir_facerestore_models = os.path.join(models_dir, "facerestore_models")