kjozsa commited on
Commit
a616720
1 Parent(s): 8780e50
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -68,14 +68,14 @@ def main():
68
  question = setup("Priest, your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after!")
69
 
70
  role = target(question)
71
- for count, _ in enumerate(range(10)):
72
- with st.spinner(f"Asking {role}..."):
 
73
  actor = Actor[role]
74
  answer = ask(actor.model, actor.system_prompt, actor.pre_prompt, question)
75
  st.write(f":blue[{actor.role} says:] {answer}")
76
  question = sanitize(answer)
77
  role = target(question)
78
- count += 1
79
 
80
 
81
  def target(question):
 
68
  question = setup("Priest, your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after!")
69
 
70
  role = target(question)
71
+ max_steps = 10
72
+ for step, _ in enumerate(range(max_steps), start=1):
73
+ with st.spinner(f"{step}/{max_steps} Asking {role}..."):
74
  actor = Actor[role]
75
  answer = ask(actor.model, actor.system_prompt, actor.pre_prompt, question)
76
  st.write(f":blue[{actor.role} says:] {answer}")
77
  question = sanitize(answer)
78
  role = target(question)
 
79
 
80
 
81
  def target(question):