Try to fix type error

app.py

@@ -33,7 +33,7 @@ In this demo, we showcase a method to make a trained Reinforcement Learning (RL)
 - Extract an interpretable policy expressed in symbolic form.
 
 For more information about KAN you can read the [paper](https://arxiv.org/abs/2404.19756), and check the [PyTorch official information](https://github.com/KindXiaoming/pykan).
-To follow the progress of KAN in RL you can check the repo [kanrl](https://github.com/riiswa/kanrl).
+To follow the progress of KAN in RL you can check the repo [kanrl](https://github.com/riiswa/kanrl) (you can run this app locally).
 
 [![riiswa/kanrl - GitHub](https://gh-card.dev/repos/riiswa/kanrl.svg)](https://github.com/riiswa/kanrl)
 """
@@ -46,6 +46,13 @@ if __name__ == "__main__":
 
     def load_video_and_dataset(_env_name):
         env_name = _env_name
+        if env_name in ["Swimmer-v3", "Hopper-v3", "HalfCheetah-v3", "Walker2d-v3"]:
+            gr.Warning(
+                "We're currently in the process of adding support for Mujoco environments, so the application may encounter crashes during this phase. We encourage contributors to join us in the repository https://github.com/riiswa/kanrl to assist in the development and support of other environments. Your contributions are invaluable in ensuring a robust and comprehensive framework."
+            )
+            torch.set_default_dtype(torch.float64)
+        else:
+            torch.set_default_dtype(torch.float32)
         agent = "ppo"
         if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
             agent ="trpo"