hello

app.py

@@ -57,7 +57,7 @@ with gr.Blocks() as demo:
             option = gr.Dropdown(choices=["Torch Tensor of 0's", "Running Average"], type="index")
             baselines = gr.Slider(label="Number of Baseline Iterations", interactive=True, minimum=0, maximum=100, value=10, step=5, info="Baseline inputs to collect for the average", render=True)
         gr.Button("ATTRIBUTE").click(fn=generate_output, inputs=[baselines, option])
-        slider = gr.Slider(label="Key Frame", minimum=0, maximum=20000, step=1, value=0)
+        slider = gr.Slider(label="Key Frame", minimum=0, maximum=300, step=1, value=0)
 
         gr.Interface(fn=get_frame_and_attribution, inputs=slider, live=True, outputs=[gr.Image(), gr.Label()])
 

train.py

@@ -204,7 +204,7 @@ class TrainingLoop:
         score_history = []
         frames = []
 
-        for i in range(1):
+        for i in range(10):
             done = False
             score = 0
             obs, _ = self.env.reset()