Update app.py

app.py

@@ -49,12 +49,9 @@ def get_frame_and_attribution(slider_value):
     attribution = display_softmax(attributions[slider_value])
 
     return frame, attribution
-
-with gr.Blocks() as demo:
-    gr.Markdown("# Introspection in Deep Reinforcement Learning")
-
+    
     with gr.Tab(label="Attribute"):
-        env_spec = gr.Textbox(label="Environment Specification (e.g.: LunarLander-v2)", lines=1)
+        env_spec = gr.Dropdown(choices=["LunarLander-v2"],type="value",multiselect=False, label="Environment Specification (e.g.: LunarLander-v2)", lines=1)
         env = gr.Interface(title="Create the Environment", allow_flagging="never", inputs=env_spec, fn=create_training_loop, outputs=gr.JSON())
 
         with gr.Row():
@@ -63,7 +60,6 @@ with gr.Blocks() as demo:
         gr.Button("ATTRIBUTE").click(fn=generate_output, inputs=[baselines, option])
         slider = gr.Slider(label="Key Frame", minimum=0, maximum=1000, step=1, value=0)
 
-        gr.Interface(fn=get_frame_and_attribution, inputs=slider, live=True, outputs=[gr.Image(), gr.Label()])
-
+        gr.Interface(fn=get_frame_and_attribution, inputs=slider, live=True, outputs=[gr.Image(label="Timestep")])
 
 demo.launch()