Skip to content

Commit

Permalink
Merge pull request #21 from scattter/fix/ollama-model-init-error
Browse files Browse the repository at this point in the history
fix ollama model init error
  • Loading branch information
itsOwen committed Sep 10, 2024
2 parents 1008610 + 2cb4206 commit e3c19a6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

1 comment on commit e3c19a6

@itsOwen
Copy link
Owner Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks :)

Please sign in to comment.