Update model_wrapper/frida_wrapper.py

model_wrapper/frida_wrapper.py

@@ -1,12 +1,11 @@
 # Захреначьте импорты сюда и сделайте метод call, как в других обёртках
 from FRIDA.model import load_model
 from FRIDA.model import generate_resp
-import os
 
 class FridaWrapper(object):
-    chkp_dir = 'Cricles/demo_detoxi/FRIDA'
+    chkp_dir = 'FRIDA/'
     def __init__(self) -> None:
-        self.model, self.tokenizer=load_model(os.path.join(FridaWrapper.chkp_dir, "classifier_head.pth"))
+        self.model, self.tokenizer=load_model(FridaWrapper.chkp_dir+"classifier_head.pth")
     
     def __call__(self, text: str) -> str:
         return generate_resp(text)