Update Space

app.py

@@ -4,7 +4,6 @@ from huggingface_hub import HfApi, login
 import os
 import yaml
 from loguru import logger
-from litrl.env.typing import EnvId
 
 SPACE_REPO = "c-gohlke/litrl"
 SPACE_REPO_TYPE = "space"
@@ -48,7 +47,7 @@ def run_demo() -> None:
     for env in environments:
         gif_paths[env] = get_best_gif(env)
     
-    def get_image(env_id: EnvId)-> gr.Image|None:
+    def get_image(env_id: str)-> gr.Image|None:
         if env_id not in gif_paths:
             logger.error(f"Environment {env_id} not found in {gif_paths}")
             return None