aryan2178/text2sql

1import gradio as gr2from transformers import GPT2Tokenizer, GPT2LMHeadModel3 4model_name = "EleutherAI/gpt-neo-125M"5model = GPT2LMHeadModel.from_pretrained(model_name,ignore_mismatched_sizes=True)6tokenizer = GPT2Tokenizer.from_pretrained(model_name)7 8 9def generate_sql(text):10    inputs = tokenizer.encode("translate English to SQL: " + text, return_tensors="pt", max_length=512, truncation=True)11    outputs = model.generate(inputs, max_length=128, num_return_sequences=1, early_stopping=True)12    sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)13 14    return sql_query15 16input_text = gr.inputs.Textbox(lines=7, label="Input Text")17output_text = gr.outputs.Textbox(label="Generated SQL Query")18 19gr.Interface(20    fn=generate_sql,21    inputs=input_text,22    outputs=output_text,23    title="Text to SQL Query Generator",24    description="Enter a natural language query and get the corresponding SQL query.",25    theme="compact"26).launch()27