File size: 1,302 Bytes
8951179 4517db3 1048a1a 4517db3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | ---
license: apache-2.0
---
# Fine-Tuned BART Model for Translation
This repository contains a fine-tuned version of the BART model for translation tasks. The model is trained to translate human instructions into CLI (Command Line Interface) commands.
## Model Details
- Model: BART
- Architecture: `facebook/bart-large`
- Fine-Tuned on: Custom translation dataset
- Max Input Length: 128 tokens
- Max Output Length: 50 tokens
- Beam Search: 4 beams
## Usage
You can use this model to generate CLI commands from human instructions.
You can either directly use the model for inference or integrate it into applications
using the provided Gradio interface.
### Inference
To perform inference using the model, you can load it.
### Gradio Interface
An interactive Gradio interface is provided for easy model interaction.
You can run the interface by executing the code in `gradio_app.py`.
The interface allows you to enter human instructions and get corresponding generated CLI commands.
## Model Files
- `config.json`: Model configuration file
- `pytorch_model.bin`: Model weights
- `tokenizer.json`: Tokenizer configuration file
- `vocab.txt`: Vocabulary file
## Acknowledgments
The initial BART model and tokenizer are from the Hugging Face Transformers library (`facebook/bart-large`).
|