Update app.py

app.py

@@ -25,7 +25,7 @@ def generate_vis(df, mode='Document'):
   for row in df.itertuples(index=False):
     if mode == 'Query':
       tok_scores = row.query_toks
-      orig_tokens = factory_max.tokenizer.tokenize(row.text)
+      orig_tokens = factory_max.tokenizer.tokenize(row.query)
       max_score = max(tok_scores.values())
       id = row.qid
     else: