Update update_embeddings.py

update_embeddings.py

@@ -30,8 +30,8 @@ start = time()
 ################################################################################
 # Configuration
 
-# Year to update embeddings for, get and set the current year
-year = str(datetime.now().year)[2:]
+# Get current year
+year = str(datetime.now().year)
 
 # Flag to force download and conversion even if files already exist
 FORCE = True
@@ -216,7 +216,7 @@ except Exception as e:
     print(f"Errored out with: {e}")
     print(f"No previous embeddings found for year: {year}")
     print("Creating new embeddings for all papers")
-    previous_embeddings = pd.DataFrame(columns=['id', 'vector', '$meta'])
+    previous_embeddings = pd.DataFrame(columns=['id', 'vector', 'title', 'abstract', 'authors', 'categories', 'month', 'year', 'url'])
 
 ########################################
 # Embed the new abstracts