tonko22 commited on
Commit
e018c03
·
1 Parent(s): a8e40a3
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,7 +36,8 @@ def main():
36
 
37
  # If using Ollama, we need to specify the API base URL
38
  # Initialize the LLM model based on configuration
39
- model_id = get_model_id(use_local=is_test)
 
40
  logger.info(f"Initializing with model: {model_id}")
41
  if is_test:
42
  api_base = get_ollama_api_base()
 
36
 
37
  # If using Ollama, we need to specify the API base URL
38
  # Initialize the LLM model based on configuration
39
+ model_id = get_model_id(use_local=is_test, provider="gemini")
40
+
41
  logger.info(f"Initializing with model: {model_id}")
42
  if is_test:
43
  api_base = get_ollama_api_base()