XufengDuan commited on
Commit
a6b7260
1 Parent(s): 9ba63d6

update scripts

Browse files
Files changed (1) hide show
  1. src/backend/model_operations.py +2 -2
src/backend/model_operations.py CHANGED
@@ -178,7 +178,7 @@ class SummaryGenerator:
178
  if 'Stimuli-2' in df_sheet.columns:
179
  Stimuli_2_column = df_sheet["Stimuli-2"]
180
 
181
- for j, prompt_value in enumerate(tqdm(prompt_column, desc=f"Processing {sheet_name}"), start=0):
182
  if exit_outer_loop:
183
  break
184
  ID = 'E' + str(i)
@@ -187,7 +187,7 @@ class SummaryGenerator:
187
  # print(ID, q_ID, prompt_value)
188
  system_prompt = envs.SYSTEM_PROMPT
189
  _user_prompt = prompt_value
190
- for ii in range(10):
191
  # user_prompt = f"{envs.USER_PROMPT}\nPassage:\n{_source}"
192
  while True:
193
  try:
 
178
  if 'Stimuli-2' in df_sheet.columns:
179
  Stimuli_2_column = df_sheet["Stimuli-2"]
180
 
181
+ for j, prompt_value in enumerate(tqdm(prompt_column[0:2], desc=f"Processing {sheet_name}"), start=0):
182
  if exit_outer_loop:
183
  break
184
  ID = 'E' + str(i)
 
187
  # print(ID, q_ID, prompt_value)
188
  system_prompt = envs.SYSTEM_PROMPT
189
  _user_prompt = prompt_value
190
+ for ii in range(1):
191
  # user_prompt = f"{envs.USER_PROMPT}\nPassage:\n{_source}"
192
  while True:
193
  try: