Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -115,16 +115,19 @@ with gr.Blocks(css="style.css", fill_height=True) as demo:
|
|
115 |
value=1.2,
|
116 |
)
|
117 |
|
118 |
-
|
|
|
119 |
generate,
|
120 |
inputs=[text_box, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
|
121 |
outputs=[text_box],
|
122 |
)
|
|
|
|
|
123 |
stop_button.click(
|
124 |
None,
|
125 |
None,
|
126 |
None,
|
127 |
-
cancels=[
|
128 |
)
|
129 |
|
130 |
if __name__ == "__main__":
|
|
|
115 |
value=1.2,
|
116 |
)
|
117 |
|
118 |
+
# Set up the generation event
|
119 |
+
generation_event = complete_button.click(
|
120 |
generate,
|
121 |
inputs=[text_box, max_new_tokens, temperature, top_p, top_k, repetition_penalty],
|
122 |
outputs=[text_box],
|
123 |
)
|
124 |
+
|
125 |
+
# Set up the stop event
|
126 |
stop_button.click(
|
127 |
None,
|
128 |
None,
|
129 |
None,
|
130 |
+
cancels=[generation_event]
|
131 |
)
|
132 |
|
133 |
if __name__ == "__main__":
|