teaevo commited on
Commit
b15f680
·
1 Parent(s): f6d8df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -97,14 +97,14 @@ def predict(input, history=[]):
97
  return response, history
98
 
99
 
100
- def sqlquery(input, conversation_history):
101
 
102
  #input_text = " ".join(conversation_history) + " " + input
103
  sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
104
  sql_outputs = sql_model.generate(**sql_encoding)
105
  sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
106
 
107
- #global conversation_history
108
 
109
  # Maintain the conversation history
110
  conversation_history.append("User: " + input)
@@ -129,7 +129,7 @@ sql_interface = gr.Interface(
129
  fn=sqlquery,
130
  theme="default",
131
  #inputs=gr.Textbox(prompt="You:"),
132
- inputs=["text", "text"],
133
  outputs=gr.Textbox(),
134
  live=True,
135
  capture_session=True,
 
97
  return response, history
98
 
99
 
100
+ def sqlquery(input):
101
 
102
  #input_text = " ".join(conversation_history) + " " + input
103
  sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
104
  sql_outputs = sql_model.generate(**sql_encoding)
105
  sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
106
 
107
+ global conversation_history
108
 
109
  # Maintain the conversation history
110
  conversation_history.append("User: " + input)
 
129
  fn=sqlquery,
130
  theme="default",
131
  #inputs=gr.Textbox(prompt="You:"),
132
+ inputs=gr.Textbox(prompt="You:"),
133
  outputs=gr.Textbox(),
134
  live=True,
135
  capture_session=True,