SQL_Generation / README.md
Canstralian's picture
Update README.md
4a0e67c verified

A newer version of the Gradio SDK is available: 5.15.0

Upgrade
metadata
title: SQL Generation
emoji: 🦀
colorFrom: red
colorTo: gray
sdk: gradio
sdk_version: 5.9.1
app_file: app.py
pinned: false
license: mit

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 datasets

Usage

After installing the dependencies, you can start the application by running:

python app.py

This 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.