beingpraveen/streamlit_text_to_sql
1
1import streamlit as st2import layer3 4model = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-english-to-sql').get_train()5tokenizer = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-tokenizer').get_train()6 7def convert(query):8 inputs = tokenizer.encode(f"translate English to SQL: {query}", return_tensors="pt")9 outputs = model.generate(inputs, max_length=1024)10 sql = tokenizer.decode(outputs[0], skip_special_tokens=True)11 return sql12 13query = st.text_input("Enter Text here", value="")14output = convert(query)15st.text_area(label="Output Sql Query:", value=output, height=100)16 