Spaces:

VamooseBambel
/

Sana-1.6B

Running on Zero

VamooseBambel commited on 28 days ago

Commit

e8ec659

•

1 Parent(s): 8913faa

Huh?

Files changed (1) hide show

nsfw_detector.py CHANGED Viewed

@@ -6,7 +6,7 @@ import numpy as np
 class NSFWDetector:
     def __init__(self):
-        self.model_path = "MichalMlodawski/nsfw-image-detection-large"
         self.feature_extractor = AutoProcessor.from_pretrained(self.model_path)
         self.model = FocalNetForImageClassification.from_pretrained(self.model_path)
         self.model.eval()

 class NSFWDetector:
     def __init__(self):
+        self.model_path = "TostAI/nsfw-image-detection-large"
         self.feature_extractor = AutoProcessor.from_pretrained(self.model_path)
         self.model = FocalNetForImageClassification.from_pretrained(self.model_path)
         self.model.eval()