fix client

app.py

@@ -106,16 +106,19 @@ class ContextualQA:
         self.client = None
 
 
-client = anthropic.Client(api_key=os.environ["ANTHROPIC_API_KEY"])
-
-
 def load_context(paper_id):
     latex_source = download_arxiv_source(paper_id)
+    client = anthropic.Client(api_key=os.environ["ANTHROPIC_API_KEY"])
     model = ContextualQA(client, model="claude-v1.3-100k")
     model.load_text(latex_source)
     return (
         model,
-        [(f"Load the paper with id {paper_id}.", "Paper loaded, Now ask a question.")],
+        [
+            (
+                f"Load the paper with id {paper_id}.",
+                "Paper loaded, You can now ask questions.",
+            )
+        ],
     )
 
 
@@ -125,6 +128,9 @@ def answer_fn(model, question, chat_history):
         chat_history.append(("No Question Asked", "Please ask a question."))
         return model, chat_history, ""
 
+    client = anthropic.Client(api_key=os.environ["ANTHROPIC_API_KEY"])
+    model.client = client
+
     response = model.ask_question(question)
 
     chat_history.append((question, response[0]["completion"]))
@@ -145,7 +151,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     with gr.Column():
         with gr.Row():
-            paper_id_input = gr.Textbox(label="Enter Paper ID", value="2303.10130")
+            paper_id_input = gr.Textbox(label="Enter Paper ID", value="2108.07258")
             btn_load = gr.Button("Load Paper")
             qa_model = gr.State()