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()
|