Update app.py

app.py

@@ -1,3 +1,16 @@
 import gradio as gr
 
-gr.load("models/NumbersStation/nsql-llama-2-7B").launch()
+from transformers import pipeline
+
+pipe = pipeline("translation", model="models/NumbersStation/nsql-llama-2-7B")
+
+def predict(text):
+  return pipe(text)[0]["translation_text"]
+
+demo = gr.Interface(
+  fn=predict,
+  inputs='text',
+  outputs='text',
+)
+
+demo.launch()