only arxiv

app.py

@@ -2,6 +2,8 @@ import json
 import os
 import shutil
 import requests
+import langchain
+from langchain.document_loaders import ArxivLoader
 
 import gradio as gr
 from huggingface_hub import Repository, InferenceClient
@@ -12,13 +14,7 @@ BOT_NAME = "Falcon"
 
 STOP_SEQUENCES = ["\nUser:", "<|endoftext|>", " User:", "###"]
 
-EXAMPLES = [
-    ["Hey Falcon! Any recommendations for my holidays in Abu Dhabi?"],
-    ["What's the Everett interpretation of quantum mechanics?"],
-    ["Give me a list of the top 10 dive sites you would recommend around the world."],
-    ["Can you tell me more about deep-water soloing?"],
-    ["Can you write a short tweet about the release of our latest AI model, Falcon LLM?"]
-    ]
+EXAMPLES = ['2309.11495', '2306.01116']
 
 client = InferenceClient(
     API_URL,
@@ -57,6 +53,12 @@ def generate(
         seed=seed,
     )
     seed = seed + 1
+
+    arxiv_key = prompt.strip().split(".")
+    if len(arxiv_key) != 2:
+        output = "Please enter a valid arXiv ID."
+        return output
+    
     formatted_prompt = format_prompt(prompt, history, system_prompt)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)