updated the filess

app.py

@@ -1,19 +1,17 @@
 import gradio as gr
-import layer
+from transformers import AutoTokenizer, AutoModelWithLMHead
 
-model = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-english-to-sql').get_train()
-tokenizer = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-tokenizer').get_train()
+tokenizer = AutoTokenizer.from_pretrained("dbernsohn/t5_wikisql_en2SQL")
+model = AutoModelWithLMHead.from_pretrained("dbernsohn/t5_wikisql_en2SQL")
 
 def greet(query):
-    input_ids = tokenizer.encode(f"translate English to SQL: {query}", return_tensors="pt")
-    outputs = model.generate(input_ids, max_length=1024)
-    sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return sql
-
+    input_text = f"translate English to Sql: {query} </s>"
+    features = tokenizer([input_text], return_tensors='pt')
+    output = model.generate(input_ids=features['input_ids'], 
+                        attention_mask=features['attention_mask'])
+    return tokenizer.decode(output[0])
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text", examples=[
-    "Show me the average price of wines in Italy by provinces",
-    "Cars built after 2020 and manufactured in Italy",
-    "Top 10 cities by their population"
+    "what are the names of all the people in the USA"
 ])
 iface.launch()

requirements.txt

@@ -1,3 +1,2 @@
-layer==0.9.350435
-torch==1.11.0
-sentencepiece==0.1.96
+transformers
+sentencepiece