gymnasium dependency

app.py

@@ -16,12 +16,11 @@ default_model_name = "gia-project/gia2-small-untrained"
 
 
 def generate_text(model_name, input_text):
-
-
     # Check if the selected model is already loaded
     if model_name not in model_pipelines:
         # Inform the user that the model is loading
         yield "Loading model..."
+
         # Load the model and create a pipeline if it's not already loaded
         generator = pipeline("text-generation", model=model_name, trust_remote_code=True)
         model_pipelines[model_name] = generator

requirements.txt

@@ -1,3 +1,4 @@
 transformers
 torch
-huggingface_hub
+huggingface_hub
+gymnasium