diff --git a/recipes/ollama/streamlit_app.py b/recipes/ollama/streamlit_app.py index dbc739f..ead887c 100644 --- a/recipes/ollama/streamlit_app.py +++ b/recipes/ollama/streamlit_app.py @@ -20,7 +20,7 @@ def extract_model_names(models_info: list) -> tuple: A tuple containing the model names. """ - return tuple(model["name"] for model in models_info["models"]) + return tuple(model["model"] for model in models_info["models"]) def main(): @@ -83,4 +83,4 @@ def main(): if __name__ == "__main__": - main() \ No newline at end of file + main()