diff --git a/src/app.py b/src/app.py index 1ea3e06..459ab0a 100644 --- a/src/app.py +++ b/src/app.py @@ -14,7 +14,7 @@ with gr.Blocks() as demo: with gr.Row(): provider = gr.Dropdown( - choices=["mock", "openai", "anthropic", "google", "deepseek"], + choices=["mock", "openai", "anthropic", "google", "deepseek", "ollama"], value="mock", label="Modello da usare" ) diff --git a/src/app/agents/predictor_agent.py b/src/app/agents/predictor_agent.py index f8f8011..6609882 100644 --- a/src/app/agents/predictor_agent.py +++ b/src/app/agents/predictor_agent.py @@ -14,7 +14,7 @@ class PredictorAgent: def __init__(self): # Ollama via HTTP locale self.providers = { - "ollama": {"type": "ollama", "host": os.getenv("OLLAMA_HOST", "http://localhost:11434"), "model": "llama3"} + "ollama": {"type": "ollama", "host": os.getenv("OLLAMA_HOST", "http://localhost:11434"), "model": "gpt-oss:latest"} } # OpenAI