Chemically-motivated/SQL_Generation
SQL Generation ๐ฆ
Welcome to the SQL Generation Gradio application! This tool leverages advanced machine learning models to assist in generating SQL queries based on natural language inputs. Whether you're a developer, data analyst, or just curious about SQL, this app aims to simplify the process of crafting SQL queries.
Features
- Natural Language to SQL: Convert plain English descriptions into SQL queries.
- Multiple Datasets: Trained on diverse datasets to handle various SQL generation tasks.
- User-Friendly Interface: Built with Gradio for an intuitive and interactive experience.
Installation
To run this application locally, ensure you have Python 3.10 or higher installed. Then, install the required dependencies:
pip install gradio transformers datasetsUsage
After installing the dependencies, you can start the application by running:
python app.pyThis will launch a local server. Open your browser and navigate to http://127.0.0.1:7860 to access the interface.
Datasets Used
The model has been trained on the following datasets:
- b-mc2/sql-create-context: Provides context for SQL query generation.
- TuneIt/o1-python: Offers examples of Python code snippets.
- HuggingFaceFW/fineweb-2: Includes various language models for fine-tuning.
- sentence-transformers/embedding-training-data: Supplies data for training sentence embeddings.
Model
The application utilizes the distilbert-base-uncased model from Hugging Face, known for its efficiency and performance in natural language processing tasks.
License
This project is licensed under the MIT License.
Acknowledgments
- Gradio for providing an easy-to-use interface for machine learning models.
- Hugging Face for hosting the pre-trained models and datasets.
- Datasets for offering a wide range of datasets for training and evaluation.
For more information, refer to the Gradio documentation.
