import streamlit as st | |
from transformers import pipeline | |
query = st.text_input("Query", value="feel good story since yesterday") | |
nlp = pipeline("text2text-generation", model="NeuML/t5-small-txtsql") | |
text = nlp(f"translate English to SQL: {query}", max_length=512)[0]["generated_text"] | |
st.write(text.replace('$=', '<=')) | |