You can deploy this model and start earning money today!

Jaanhavi/
my-wildguard-roberta

Name: my-wildguard-roberta
Availability: InStock

Views

No views yet

Model Card for Model ID

Model Details

To load:

python

from transformers import RobertaForSequenceClassification, RobertaTokenizer

model = RobertaForSequenceClassification.from_pretrained("Jaanhavi/my-wildguard-roberta")
tokenizer = RobertaTokenizer.from_pretrained("Jaanhavi/my-wildguard-roberta")

To make:

python

text = "Your prompt here"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
prediction = outputs.logits.argmax(dim=-1).item()
print("Harmful" if prediction == 1 else "Unharmful")