Update app.py

app.py

@@ -3,8 +3,8 @@ import torch
 import transformers
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 
-tokenizer = AutoTokenizer.from_pretrained('gbarone77/t5-small-finetuned-wikisql-with-cols')
-model = T5ForConditionalGeneration.from_pretrained('gbarone77/t5-small-finetuned-wikisql-with-cols')
+tokenizer = AutoTokenizer.from_pretrained('gbarone77/t5flan-small-finetuned-wikisql-with-cols')
+model = T5ForConditionalGeneration.from_pretrained('gbarone77/t5flan-small-finetuned-wikisql-with-cols')
 
 def translate_to_sql(text):
     inputs = tokenizer(text, padding='longest', max_length=64, return_tensors='pt')