title | emoji | colorFrom | colorTo | sdk | sdk_version | app_file | pinned | short_description |
---|---|---|---|---|---|---|---|---|
PhishingEmailEvaluation |
📈 |
blue |
pink |
gradio |
5.14.0 |
app.py |
false |
Phishing Email Evaluation |
This project demonstrates a phishing detection application powered by Transformers and served via a Gradio web interface. The project includes:
- Data Generation: A script to generate sample CSV datasets.
- Model Training: Fine-tuning a Transformer model on a custom phishing dataset.
- Web App: A Gradio app with multiple tabs to evaluate URLs and emails.
-
Install Dependencies:
pip install -r requirements.txt
-
Generate Sample Data:
python generate_csv.py
-
Train the Model:
python train.py
-
Run the Web App:
python app.py
Open the provided URL in your browser to interact with the app.
- Adjust the dataset and model parameters as needed for your use-case.
- This project uses DistilBERT as a base model; you can swap it with another model if required.