Update model_wrapper/frida_wrapper.py

model_wrapper/frida_wrapper.py

@@ -8,4 +8,4 @@ class FridaWrapper(object):
         self.model, self.tokenizer=load_model(FridaWrapper.chkp_dir+"classifier_head.pth")
     
     def __call__(self, text: str) -> str:
-        return generate_resp(text)
+        return generate_resp(text,self.model,self.tokenizer)