# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("rocker417/malicious-url-detection")
model = AutoModelForSequenceClassification.from_pretrained("rocker417/malicious-url-detection", device_map="auto")Quick Links
Malicious-Url-Detection
Using this model, you can detects harmful links created to harm people such as phishing, malware urls. this model Classifies if the urls addresses are malware and benign. Type the domain name of the url address in the text field for classification in API: Like this: "huggingface.com"
This model is a fine-tuned version of distilbert-base-uncased on an kmack/Phishing_urls dataset.
- Downloads last month
- 9
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="rocker417/malicious-url-detection")