Spaces:

SengTak
/

LEC_HarmfulPromptClassifier

Sleeping

SengTak commited on Dec 31, 2024

Commit

f0e1736

1 Parent(s): 0a98cc1

Update space fixed l3prune import

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import sys
 import os
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
-from l3prune import LLMEncoder
 #load the model
 best_clf = joblib.load("./saved/classifier_llama32.joblib")

 import sys
 import os
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+from .l3prune import LLMEncoder
 #load the model
 best_clf = joblib.load("./saved/classifier_llama32.joblib")