davidberenstein1957 HF staff commited on
Commit
da92b2b
·
verified ·
1 Parent(s): b36a3f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def run_query(hub_repo_id: str, query: str, split: str, column: str):
74
  SELECT *
75
  FROM df
76
  ORDER BY array_cosine_distance(embeddings, {vector.tolist()}::FLOAT[256])
77
- LIMIT 5
78
  """
79
  ).to_df()
80
  return gr.Dataframe(df_results, visible=True)
@@ -137,7 +137,7 @@ with gr.Blocks() as demo:
137
 
138
  btn_run = gr.Button("Search", visible=False)
139
 
140
- results_output = gr.Dataframe(label="Results", visible=False)
141
 
142
  search_in.submit(get_iframe, inputs=search_in, outputs=search_out).then(
143
  fn=load_dataset_from_hub,
 
74
  SELECT *
75
  FROM df
76
  ORDER BY array_cosine_distance(embeddings, {vector.tolist()}::FLOAT[256])
77
+ LIMIT 20
78
  """
79
  ).to_df()
80
  return gr.Dataframe(df_results, visible=True)
 
137
 
138
  btn_run = gr.Button("Search", visible=False)
139
 
140
+ results_output = gr.Dataframe(label="Results", visible=False, wrap=True)
141
 
142
  search_in.submit(get_iframe, inputs=search_in, outputs=search_out).then(
143
  fn=load_dataset_from_hub,