Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import requests
 from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
 from stable_baselines3 import PPO
+import gym  # Ensure gym is imported
 from gym import spaces
 from stable_baselines3.common.vec_env import DummyVecEnv
 import plotly.graph_objects as go
@@ -113,3 +114,4 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
+