Spaces:
Sleeping
Sleeping
prov upd
Browse files
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()
|