Update app.py

app.py

@@ -1,12 +1,13 @@
 import arxiv
 import gradio as gr
-from langchain.document_loaders import OnlinePDFLoader
+from langchain.document_loaders import OnlinePDFLoader, PyPDFLoader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.llms import HuggingFaceHub
 from langchain.embeddings import HuggingFaceHubEmbeddings
 from langchain.vectorstores import Chroma
 from langchain.chains import RetrievalQA
 
+repo_id = 'mistralai/Mistral-7B-v0.1'
 client = arxiv.Client()
 
 def loading_paper(): return 'Loading...'
@@ -20,7 +21,7 @@ def paper_changes(paper_id):
     embeddings = HuggingFaceHubEmbeddings()
     db = Chroma.from_documents(texts, embeddings)
     retriever = db.as_retriever()
-    llm = HuggingFaceHub(repo_id='mistralai/Mistral-7B-v0.1', model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096})
+    llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096})
     global qa
     qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=retriever, return_source_documents=True)
     return 'Ready!!'