WandbFluencyModel.predict:v0
Version
0
Last updated
Last updated by
Calls:
1
2
3
4
5
6
7
import weave
@weave.op()
async def predict(self, text: str):
pipeline_output = self._pipeline(text)
return {"flagged": pipeline_output[0]["label"] == "LABEL_0"}