Update app.py

app.py

@@ -18,8 +18,8 @@ from urllib.parse import urljoin
 
 
 def initialize(arxivcode):  
-    loader = ArxivLoader(query=str(arxivcode),)
-    docs = loader.load()
+    #loader = ArxivLoader(query=str(arxivcode),)
+    #docs = loader.load()
     #retriever = ArxivRetriever(
     #    load_max_docs=2,
     #    get_full_documents=True,
@@ -28,6 +28,12 @@ def initialize(arxivcode):
     #for i in range(len(docs)): 
     #    docs[i].metadata['Published'] = str(docs[i].metadata['Published'])
 
+    # Load, chunk and index the contents of the blog.
+    url = ['https://arxiv.org/abs/%s' % arxivcode]
+    loader = WebBaseLoader(url)
+    docs = loader.load()
+
+
     # LLM model
     rate_limiter = InMemoryRateLimiter(
         requests_per_second=0.1,  # <-- MistralAI free. We can only make a request once every second