Zekun Wu commited on
Commit
6d335d2
·
1 Parent(s): 234c1c2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -92,8 +92,8 @@ def get_potshots(n_repeat=1, batch=20, role="Developer", tone="humorous", audien
92
  if len(total_potshots) > desired_count:
93
  total_potshots = total_potshots[:desired_count]
94
 
95
- prompt_list = [potshot for potshot in total_potshots]
96
- return {"potshots": prompt_list}
97
 
98
 
99
  # Streamlit App Interface
 
92
  if len(total_potshots) > desired_count:
93
  total_potshots = total_potshots[:desired_count]
94
 
95
+ df = pd.DataFrame([{"potshot": potshot, "role": role, "tone": tone, "audience": audience, "values": values} for potshot in total_potshots])
96
+ return df
97
 
98
 
99
  # Streamlit App Interface