Update src/main.py

src/main.py

@@ -49,7 +49,7 @@ def split_data(data):
 def ingest_chunks(chunks):
     embedding = OllamaEmbeddings(
     base_url='https://thewise-ollama-server.hf.space',
-    model="nomic-embed-text:latest",
+    model="nomic-embed-text",
 )
     vector_store = DocArrayInMemorySearch.from_documents(chunks, embedding)
 
@@ -64,7 +64,7 @@ def retreival(vector_store, k):
     #Creating LLM
     llm = ChatOllama(
     base_url='https://thewise-ollama-server.hf.space',
-    model="codellama:34b")
+    model="codellama")
                      
     # Define the system message template
     #Adding CHAT HISTORY to the System template explicitly because mainly Chat history goes to Condense the Human Question with Backround (Not template), but System template goes straight the LLM Chain