Update app.py

app.py

@@ -225,7 +225,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
 
     with gr.Group():
         model_chooser = gr.Radio(label='Choose Your Model', choices=list(model_info.keys()), value=model_name)
-        load_on_gpu = gr.Checkbox(label='Load on GPU', value=True)
+        load_on_gpu = gr.Checkbox(label='Load on GPU', visible=False, value=True)
         welcome_model = gr.Markdown(welcome_message.format(model_name=model_name))
     with gr.Blocks() as demo_main:
         gr.Markdown('## The Prompt to Analyze')        
@@ -270,7 +270,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
         with gr.Row():
             for btn in tokens_container:
                 btn.render()
-        use_gpu = gr.Checkbox(label='Use GPU', value=True)
+        use_gpu = gr.Checkbox(label='Use GPU', visible=True, value=True)
         
         progress_dummy = gr.Markdown('', elem_id='progress_dummy')
         interpretation_bubbles = [gr.Textbox('', container=False, visible=False) for i in range(MAX_NUM_LAYERS)]