Update model_wrapper/frida_wrapper.py

model_wrapper/frida_wrapper.py

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