ping98k commited on
Commit
c29b692
·
1 Parent(s): 753833c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import os, json, gradio as gr
2
  from concurrent.futures import ThreadPoolExecutor, as_completed
3
  from tqdm import tqdm
@@ -146,7 +148,7 @@ demo = gr.Interface(
146
  gr.Textbox(lines=1, label="Criteria (comma separated)"),
147
  gr.Number(value=NUM_GENERATIONS_DEFAULT, label="Number of Generations"),
148
  gr.Number(value=NUM_TOP_PICKS_DEFAULT, label="Top Picks (k)"),
149
- gr.Number(value=POOL_SIZE_DEFAULT, label="Pool Size"),
150
  gr.Number(value=MAX_WORKERS_DEFAULT, label="Max Workers")
151
  ],
152
  outputs=gr.Textbox(label="Top picks")
 
1
+ from dotenv import load_dotenv
2
+ load_dotenv()
3
  import os, json, gradio as gr
4
  from concurrent.futures import ThreadPoolExecutor, as_completed
5
  from tqdm import tqdm
 
148
  gr.Textbox(lines=1, label="Criteria (comma separated)"),
149
  gr.Number(value=NUM_GENERATIONS_DEFAULT, label="Number of Generations"),
150
  gr.Number(value=NUM_TOP_PICKS_DEFAULT, label="Top Picks (k)"),
151
+ gr.Number(value=POOL_SIZE_DEFAULT, label="Filter Size"),
152
  gr.Number(value=MAX_WORKERS_DEFAULT, label="Max Workers")
153
  ],
154
  outputs=gr.Textbox(label="Top picks")