import streamlit as st

from transformers import pipeline

query = st.text_input("Query")

nlp = pipeline("text2text-generation", model="t5-small-txtsql")
text = nlp(f"translate English to SQL: {query}", max_length=512)[0]["generated_text"]
st.write(text.replace("$=", "<="))