App changed

app.py

@@ -1,7 +1,17 @@
 import gradio as gr
+import layer
+
+query = "Show me the average price of wines in Italy by provinces"
+
+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()
+
+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
 
-def greet(name):
-    return "Hello " + name + "!!"
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface.launch()