Added variables

app.py

@@ -1,12 +1,24 @@
 import gradio as gr
 import json
 from run_llm import template_all, prompt2_pos, prompt2_chunk, prompt2_parse, demon_pos, demon_chunk, demon_parse, model_mapping
+from tqdm import tqdm
 
 # Your existing code
 theme = gr.themes.Soft()
 
+gid_list = selected_idx[args.start:args.end]
+
+ptb = []
+with open('ptb.jsonl', 'r') as f:
+    for l in f:
+        ptb.append(json.loads(l))
+        
+for gid in tqdm(gid_list, desc='Query'):
+            text = ptb[gid]['text']
+
 # Function to process text based on model and task
 def process_text(model_name, task, text):
+    
     # Define prompts for each strategy based on the task
     strategy_prompts = {
         'Strategy 1': template_all.format(text),