Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 11

Commit

b0858bb

verified ·

1 Parent(s): 8808739

Update model_wrapper/frida_wrapper.py

Files changed (1) hide show

model_wrapper/frida_wrapper.py CHANGED Viewed

@@ -4,7 +4,7 @@ from FRIDA.model import generate_resp
 import os
 class FridaWrapper(object):
-    chkp_dir = './'
     def __init__(self) -> None:
         self.model, self.tokenizer=load_model(os.path.join(FridaWrapper.chkp_dir, "classifier_head.pth"))

 import os
 class FridaWrapper(object):
+    chkp_dir = 'Cricles/demo_detoxi/FRIDA'
     def __init__(self) -> None:
         self.model, self.tokenizer=load_model(os.path.join(FridaWrapper.chkp_dir, "classifier_head.pth"))