Spaces:
Running on Zero
Running on Zero
Ben commited on
Commit ·
c533f9b
1
Parent(s): 69f399c
Use LunarLander v3 environment
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def simulate_agent(stage_selection):
|
|
| 46 |
return None, f"Weight download failed. Error: {str(e)}"
|
| 47 |
|
| 48 |
# Initialize env
|
| 49 |
-
env = gym.make("LunarLander-
|
| 50 |
|
| 51 |
# Initialize model on CPU
|
| 52 |
device = torch.device("cpu")
|
|
|
|
| 46 |
return None, f"Weight download failed. Error: {str(e)}"
|
| 47 |
|
| 48 |
# Initialize env
|
| 49 |
+
env = gym.make("LunarLander-v3", render_mode="rgb_array")
|
| 50 |
|
| 51 |
# Initialize model on CPU
|
| 52 |
device = torch.device("cpu")
|