Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 11

Commit

f478e79

verified ·

1 Parent(s): fa61f7e

Update model_wrapper/frida_wrapper.py

Files changed (1) hide show

model_wrapper/frida_wrapper.py CHANGED Viewed

@@ -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)

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