vbzvibin commited on
Commit
b1488f3
1 Parent(s): 0a825ca

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -113,8 +113,9 @@ elif nav == "Text2SQL":
113
 
114
  # text = "what interface is measure indicator code = 72_HR_ABX and version is 1 and source is TD"
115
  table_name = "df_qna"
 
116
 
117
- txt_sql = inference(question=sql_txt, table=df_qna.columns)
118
 
119
 
120
  ### SQL Modification
 
113
 
114
  # text = "what interface is measure indicator code = 72_HR_ABX and version is 1 and source is TD"
115
  table_name = "df_qna"
116
+ table_column = ['Patient_Name', 'Country', 'Disease', 'CUI', 'Snomed', 'Oxygen_Rate','Med_Type', 'Admission_Date']
117
 
118
+ txt_sql = inference(question=sql_txt, table=table_column)
119
 
120
 
121
  ### SQL Modification