Mubbashir Ahmed commited on
Commit
f34278a
·
1 Parent(s): 6725b24
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def evaluate_model(model_name, user_input, expected_sql, chat_history):
85
  # ------------------------
86
  def get_random_spider_prompt():
87
  sample = random.choice(spider_dataset)
88
- return sample["question"], sample["query"]
89
 
90
  # ------------------------
91
  # Gradio UI
 
85
  # ------------------------
86
  def get_random_spider_prompt():
87
  sample = random.choice(spider_dataset)
88
+ return sample["question"], sample["query"], sample["query"] # Return expected SQL twice
89
 
90
  # ------------------------
91
  # Gradio UI