davidmezzetti commited on
Commit
95a7058
1 Parent(s): 0d71852

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,6 +4,6 @@ from transformers import pipeline
4
 
5
  query = st.text_input("Query")
6
 
7
- nlp = pipeline("text2text-generation", model="t5-small-txtsql")
8
  text = nlp(f"translate English to SQL: {query}", max_length=512)[0]["generated_text"]
9
  st.write(text.replace("$=", "<="))
 
4
 
5
  query = st.text_input("Query")
6
 
7
+ nlp = pipeline("text2text-generation", model="NeuML/t5-small-txtsql")
8
  text = nlp(f"translate English to SQL: {query}", max_length=512)[0]["generated_text"]
9
  st.write(text.replace("$=", "<="))