Update app.py
Browse files
app.py
CHANGED
@@ -255,7 +255,7 @@ with gr.Blocks(theme=gr.themes.Glass(), css='styles.css') as demo:
|
|
255 |
with gr.Row():
|
256 |
for btn in tokens_container:
|
257 |
btn.render()
|
258 |
-
use_gpu = gr.
|
259 |
|
260 |
progress_dummy = gr.Markdown('', elem_id='progress_dummy')
|
261 |
interpretation_bubbles = [gr.Textbox('', container=False, visible=False) for i in range(MAX_NUM_LAYERS)]
|
|
|
255 |
with gr.Row():
|
256 |
for btn in tokens_container:
|
257 |
btn.render()
|
258 |
+
use_gpu = gr.Checkbox('Use GPU', value=True)
|
259 |
|
260 |
progress_dummy = gr.Markdown('', elem_id='progress_dummy')
|
261 |
interpretation_bubbles = [gr.Textbox('', container=False, visible=False) for i in range(MAX_NUM_LAYERS)]
|