Update app.py

app.py

@@ -32,7 +32,7 @@ def get_vectorstore_from_url(url):
     return vector_store
 
 def get_context_retriever_chain(vector_store):
-    llm = HuggingFaceHub(repo_id = "google/gemma-2b", model_kwargs = {"temperature":0.5, "max_length":512})
+    llm = HuggingFaceHub(repo_id = "meta-llama/Meta-Llama-3-8B", model_kwargs = {"temperature":0.5, "max_length":512})
     retriever = vector_store.as_retriever()
     
     prompt = ChatPromptTemplate.from_messages([
@@ -46,7 +46,7 @@ def get_context_retriever_chain(vector_store):
     return retriever_chain
     
 def get_conversational_rag_chain(retriever_chain): 
-    llm = HuggingFaceHub(repo_id = "google/gemma-2b", model_kwargs = {"temperature":0.5, "max_length":512})
+    llm = HuggingFaceHub(repo_id = "meta-llama/Meta-Llama-3-8B", model_kwargs = {"temperature":0.5, "max_length":512})
     
     prompt = ChatPromptTemplate.from_messages([
       ("system", "Answer the user's questions based on the below context:\n\n{context}"),
@@ -67,8 +67,7 @@ def get_response(user_input):
         "input": user_input
     })
     
-    # return response['answer']
-    return response
+    return response['answer']
 
 # app config
 st.set_page_config(page_title="Chat with websites", page_icon="🤖")