Spaces:
Sleeping
Sleeping
Commit
·
986cd70
1
Parent(s):
394c17f
Update app.py
Browse files
app.py
CHANGED
@@ -28,23 +28,8 @@ followup_message = st.sidebar.text_area("Edit Message:")
|
|
28 |
row_count = st.session_state.get("row_count", 1)
|
29 |
generate_all_edits = st.sidebar.button("Generate All Edits")
|
30 |
|
31 |
-
class WorkerThread(threading.Thread):
|
32 |
-
def __init__(self, jobs, results):
|
33 |
-
super().__init__()
|
34 |
-
self.jobs = jobs
|
35 |
-
self.results = results
|
36 |
-
|
37 |
-
def run(self):
|
38 |
-
while True:
|
39 |
-
job = self.jobs.get()
|
40 |
-
if job is None:
|
41 |
-
break
|
42 |
-
i, messages = job
|
43 |
-
result = generate_response(i, messages)
|
44 |
-
self.results.put(result)
|
45 |
-
|
46 |
if generate_all_edits:
|
47 |
-
# Creating a list of
|
48 |
messages = [st.session_state.get(f"message{i}", "") for i in range(row_count)]
|
49 |
|
50 |
jobs = Queue()
|
@@ -83,7 +68,7 @@ if generate_all_edits:
|
|
83 |
st.session_state[f"response_tokens{i}"] = response_tokens
|
84 |
st.session_state[f"word_count{i}"] = word_count
|
85 |
|
86 |
-
# ...
|
87 |
|
88 |
if add_row:
|
89 |
row_count += 1
|
|
|
28 |
row_count = st.session_state.get("row_count", 1)
|
29 |
generate_all_edits = st.sidebar.button("Generate All Edits")
|
30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
if generate_all_edits:
|
32 |
+
# Creating a list of messages
|
33 |
messages = [st.session_state.get(f"message{i}", "") for i in range(row_count)]
|
34 |
|
35 |
jobs = Queue()
|
|
|
68 |
st.session_state[f"response_tokens{i}"] = response_tokens
|
69 |
st.session_state[f"word_count{i}"] = word_count
|
70 |
|
71 |
+
# ...
|
72 |
|
73 |
if add_row:
|
74 |
row_count += 1
|