Update app.py

app.py

@@ -11,10 +11,10 @@ def generate_sql(payload):
     schema = payload.get("schema", "")
     sample_rows = payload.get("sample_rows", [])
 
-    # Convert sample rows into flat string
+    # Convert sample rows to a flat string
     rows_text = " ".join([str(row) for row in sample_rows]) if sample_rows else ""
 
-    # Construct prompt for instruction tuning
+    # Construct the prompt
     prompt = (
         f"You are a SQL expert.\n"
         f"Schema: {schema}\n"
@@ -23,9 +23,8 @@ def generate_sql(payload):
         f"Generate SQL:"
     )
 
-    # Tokenize and generate SQL
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids
-    outputs = model.generate(input_ids, max_length=256, temperature=0.6)
+    outputs = model.generate(input_ids, max_length=256, temperature=0.7)
     sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
     return sql
@@ -35,7 +34,8 @@ demo = gr.Interface(
     inputs=gr.JSON(label="JSON (question, schema, sample_rows)"),
     outputs="text",
     title="FLAN-T5 Text-to-SQL",
-    description="Using FLAN-T5 to generate SQL from natural language and tabular schema."
+    description="Using FLAN-T5 to generate SQL from natural language and tabular schema.",
+    allow_flagging="never"
 )
 
 demo.launch()