Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from langchain import HuggingFaceHub
 from langchain_community.document_loaders import WebBaseLoader, PyPDFLoader
 from langchain_community.vectorstores import Chroma
 from langchain_community import embeddings 
@@ -8,9 +9,11 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
 from langchain.output_parsers import PydanticOutputParser
 from langchain.text_splitter import CharacterTextSplitter
+import os
 
 def process_input(urls, question):
-    model_local = ChatOllama(model="llama2")
+    os.environ['HUGGINGFACEHUB_API_TOKEN'] = 'HUGGINGFACEHUB_API_TOKEN'
+    model_local = HuggingFaceHub(repo_id="google/gemma-7b", model_kwargs={"temperature":0,"max_length":64})
     # Convert string of URLs to list
     urls_list = urls.split("\n")
     docs = [WebBaseLoader(url).load() for url in urls_list]