fix ollama model init error

gmh5225 · Sep 10, 2024 · 2cb4206 · 2cb4206
1 parent 1008610
commit 2cb4206
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -171,7 +171,7 @@ def get_last_url_from_chat(messages):
 
 def initialize_web_scraper_chat(url=None):
     if st.session_state.selected_model.startswith("ollama:"):
-        model = OllamaModel(st.session_state.selected_model[7:])
+        model = st.session_state.selected_model
     else:
         model = st.session_state.selected_model