Spaces:

SengTak
/

LEC_HarmfulPromptClassifier

Sleeping

SengTak commited on Dec 31, 2024

Commit

58aa85b

verified ·

1 Parent(s): 41409b3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ best_clf = joblib.load("./saved/classifier_llama32.joblib")
 encoder = LLMEncoder.from_pretrained(
-    "./saved/pruned_encoder_llama32",
     device_map="cpu",
     torch_dtype=torch.bfloat16,
     #torch_dtype=torch,

 encoder = LLMEncoder.from_pretrained(
+    "SengTak/lec_harmfulprompclassifierencoder",
     device_map="cpu",
     torch_dtype=torch.bfloat16,
     #torch_dtype=torch,