txtsql / app.py
davidmezzetti's picture
Update app.py
df0005c
raw
history blame
323 Bytes
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('$=', '<='))