Update earnings_app.py

earnings_app.py

@@ -3,6 +3,7 @@
 import asyncio
 import os
 import openai
+import wandb
 
 from typing import List, Optional
 # from pydantic import BaseModel, Field
@@ -34,6 +35,10 @@ from llama_index import GPTVectorStoreIndex
 set_global_handler("wandb", run_args={"project": "final-project-v1"})
 wandb_callback = llama_index.global_handler
 
+run = wandb.init()
+artifact = run.use_artifact('rlpeter70/final-project-v1/earnings-index:v0', type='storage_context')
+artifact_dir = artifact.download()
+
 from dotenv import load_dotenv
 load_dotenv()