Adding model to my space

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+# Load model + tokenizer
+model_name = "tscholak/cxmefzzi"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+def text_to_sql(question, schema=""):
+    """
+    Convert natural language question into SQL query.
+    Schema can be passed as a string (table + column names).
+    """
+    # Format input as Spider expects
+    if schema:
+        prompt = f"{schema} {question}"
+    else:
+        prompt = question
+
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+    outputs = model.generate(**inputs, max_length=256)
+    sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return sql_query
+
+# Define API endpoint (no UI)
+iface = gr.Interface(
+    fn=text_to_sql,
+    inputs=[
+        gr.Textbox(label="Question"),
+        gr.Textbox(label="Schema (optional)", placeholder="table: columns, ...")
+    ],
+    outputs="text",
+)
+
+# Launch API (set share=True for external URL)
+iface.launch()

requirments.txt

@@ -0,0 +1,2 @@
+gradio
+transformers