research14 commited on
Commit
c0ac2c5
·
1 Parent(s): c8a8f0b

Added selected_idx definition

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,7 +6,11 @@ from tqdm import tqdm
6
  # Your existing code
7
  theme = gr.themes.Soft()
8
 
9
- gid_list = selected_idx[args.start:args.end]
 
 
 
 
10
 
11
  ptb = []
12
  with open('ptb.jsonl', 'r') as f:
 
6
  # Your existing code
7
  theme = gr.themes.Soft()
8
 
9
+ with open('sample_uniform_1k_2.txt', 'r') as f:
10
+ selected_idx = f.readlines()
11
+ selected_idx = [int(i.strip()) for i in selected_idx]#[s:e]
12
+
13
+ gid_list = selected_idx[0]
14
 
15
  ptb = []
16
  with open('ptb.jsonl', 'r') as f: