thewise commited on
Commit
a99c73e
1 Parent(s): 9675eb8

Update src/main.py

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -67,7 +67,7 @@ def retreival(vector_store, k):
67
  #Creating LLM
68
  llm = ChatOllama(
69
  base_url='https://thewise-ollama-server.hf.space',
70
- model="codellama",
71
  callback_manager=CallbackManager([StreamingStdOutCallbackHandler()]))
72
 
73
  # llm = Ollama(
 
67
  #Creating LLM
68
  llm = ChatOllama(
69
  base_url='https://thewise-ollama-server.hf.space',
70
+ model="llama2",
71
  callback_manager=CallbackManager([StreamingStdOutCallbackHandler()]))
72
 
73
  # llm = Ollama(