Update agent_llama_ui.py

agent_llama_ui.py

@@ -22,9 +22,11 @@ default_iterations = 2
 default_temperature = 0.5
 default_topp = 0.95
 
+MODELS_DIR = "models"
+
 @st.cache_resource
 def agent(model, temperature, top_p, context_length, load_8bit, load_4bit, max_iterations):    
-    ag = SmartAgent(f"./models/{model}" if os.path.exists(f"./models/{model}") else model, temp=temperature, top_p=top_p, load_in_4bit=load_4bit, load_in_8bit=load_8bit, ctx_len=context_length, max_iterations=max_iterations) if model else None
+    ag = SmartAgent(f"{MODELS_DIR}/{model}" if os.path.exists(f"{MODELS_DIR}/{model}") else model, temp=temperature, top_p=top_p, load_in_4bit=load_4bit, load_in_8bit=load_8bit, ctx_len=context_length, max_iterations=max_iterations) if model else None
     st.session_state["temperature_executive"] = temperature
     st.session_state["max_iterations_executive"] = max_iterations
     st.session_state["model_executive"] = model
@@ -36,7 +38,7 @@ def agent(model, temperature, top_p, context_length, load_8bit, load_4bit, max_i
     
 def get_models():
     supported_extensions = ["bin","pth","gguf"]
-    models_directory = "./models"  # Replace with the actual path
+    models_directory = f"{MODELS_DIR}"  # Replace with the actual path
     # Use os.listdir to get a list of filenames in the directory
     os.makedirs(models_directory, exist_ok=True)