ShadowTEM commited on
Commit
f51f9bc
·
verified ·
1 Parent(s): 0f3e652

added llm config variable

Browse files
Files changed (1) hide show
  1. app/llm.py +1 -1
app/llm.py CHANGED
@@ -23,7 +23,7 @@ class LLMProcessor:
23
  config_path = os.path.join(base_dir, "config", "config.yml")
24
  with open(config_path, "r") as file:
25
  self.config_data = yaml.safe_load(file)
26
-
27
  # --- Gemini Initialization ---
28
  gemini_config = llm_config.get("gemini", {})
29
  self.gemini_api_key = os.getenv("GEMINI_API_KEY") # From Hugging Face secrets
 
23
  config_path = os.path.join(base_dir, "config", "config.yml")
24
  with open(config_path, "r") as file:
25
  self.config_data = yaml.safe_load(file)
26
+ llm = self.config_data.get("llm", {})
27
  # --- Gemini Initialization ---
28
  gemini_config = llm_config.get("gemini", {})
29
  self.gemini_api_key = os.getenv("GEMINI_API_KEY") # From Hugging Face secrets