Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# Transformers Fine Tuner
|
2 |
|
3 |
Transformers Fine Tuner is a user-friendly Gradio interface that enables seamless fine-tuning of pre-trained transformer models on custom datasets.
|
@@ -12,4 +24,59 @@ Transformers Fine Tuner is a user-friendly Gradio interface that enables seamles
|
|
12 |
## Setup
|
13 |
|
14 |
1. Clone the repository:
|
15 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
sdk: gradio
|
4 |
+
colorFrom: green
|
5 |
+
colorTo: purple
|
6 |
+
title: Transformers Fine Tuner
|
7 |
+
emoji: ⚡
|
8 |
+
thumbnail: >-
|
9 |
+
https://cdn-uploads.huggingface.co/production/uploads/64fbe312dcc5ce730e763dc6/EgMfbe-SvtWnERIw6k27e.png
|
10 |
+
short_description: Transformers Fine Tuner is a user-friendly Gradio interface
|
11 |
+
---
|
12 |
+
|
13 |
# Transformers Fine Tuner
|
14 |
|
15 |
Transformers Fine Tuner is a user-friendly Gradio interface that enables seamless fine-tuning of pre-trained transformer models on custom datasets.
|
|
|
24 |
## Setup
|
25 |
|
26 |
1. Clone the repository:
|
27 |
+
```bash
|
28 |
+
git clone https://github.com/canstralian/Transformers-Fine-Tuner.git
|
29 |
+
cd Transformers-Fine-Tuner
|
30 |
+
```
|
31 |
+
|
32 |
+
2. Create and activate a virtual environment:
|
33 |
+
```bash
|
34 |
+
python -m venv venv
|
35 |
+
source venv/bin/activate # On Windows use `venv\Scripts\activate`
|
36 |
+
```
|
37 |
+
|
38 |
+
3. Install the required dependencies:
|
39 |
+
```bash
|
40 |
+
pip install -r requirements.txt
|
41 |
+
```
|
42 |
+
|
43 |
+
## Usage
|
44 |
+
|
45 |
+
To launch the Gradio interface, run the following command:
|
46 |
+
```bash
|
47 |
+
python app.py
|
48 |
+
```
|
49 |
+
|
50 |
+
## Example
|
51 |
+
|
52 |
+
1. Enter the URL of your dataset.
|
53 |
+
2. Select a pre-trained transformer model from the dropdown.
|
54 |
+
3. Adjust the training parameters such as epochs, batch size, and learning rate.
|
55 |
+
4. Click the "Submit" button to start the fine-tuning process.
|
56 |
+
|
57 |
+
## File Structure
|
58 |
+
|
59 |
+
- `app.py`: Main script to launch the Gradio interface.
|
60 |
+
- `data/preprocess.py`: Script to load and preprocess datasets.
|
61 |
+
- `.github/workflows/python-app.yml`: GitHub Actions workflow for CI/CD pipeline.
|
62 |
+
|
63 |
+
## Contributing
|
64 |
+
|
65 |
+
If you would like to contribute to this project, please fork the repository and submit a pull request.
|
66 |
+
|
67 |
+
## License
|
68 |
+
|
69 |
+
This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for more details.
|
70 |
+
|
71 |
+
## Acknowledgements
|
72 |
+
|
73 |
+
This project uses the following libraries and frameworks:
|
74 |
+
- [Gradio](https://gradio.app/)
|
75 |
+
- [Hugging Face Transformers](https://huggingface.co/transformers/)
|
76 |
+
- [Pandas](https://pandas.pydata.org/)
|
77 |
+
|
78 |
+
## Contact
|
79 |
+
|
80 |
+
For any inquiries or support, please contact the repository owner at [canstralian](https://github.com/canstralian).
|
81 |
+
|
82 |
+
---
|