stvnnnnnn commited on
Commit
3aa054d
·
verified ·
1 Parent(s): f2d0983

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -125,7 +125,7 @@ def nl2sql(q: NLQuery):
125
  query_en = text # fallback seguro
126
 
127
  # Procesar con TAPEX
128
- df = get_table_from_wikisql(SPLIT, INDEX, MAX_ROWS)
129
  enc = tok(table=df, query=query_en, return_tensors="pt", truncation=True)
130
  if torch.cuda.is_available():
131
  enc = {k: v.to("cuda") for k, v in enc.items()}
 
125
  query_en = text # fallback seguro
126
 
127
  # Procesar con TAPEX
128
+ df = get_table(SPLIT, INDEX, MAX_ROWS)
129
  enc = tok(table=df, query=query_en, return_tensors="pt", truncation=True)
130
  if torch.cuda.is_available():
131
  enc = {k: v.to("cuda") for k, v in enc.items()}