Upload folder using huggingface_hub

app.py

@@ -20,10 +20,10 @@ setup_logging()
 paper_dir = 'path_to_temp_storage'
 prompt_dir = 'iclr2024'
 api_keys = {
-    'openai_api_key': os.environ.get('openai_api_key'),
-    'claude_api_key': os.environ.get('anthropic_api_key'),
-    'gemini_api_key': os.environ.get('gemini_api_key'),
-    'commandr_api_key': os.environ.get('commandr_api_key')
+    'openai_api_key': os.environ.get('OPENAI_API_KEY'),
+    'claude_api_key': os.environ.get('ANTHROPIC_API_KEY'),
+    'gemini_api_key': os.environ.get('GEMINI_API_KEY'),
+    'commandr_api_key': os.environ.get('COMMANDR_API_KEY')
 }
 
 use_real_api = True

models.py

@@ -22,10 +22,10 @@ class PaperProcessor:
     def __init__(self, prompt_dir, model, openai_api_key, claude_api_key, gemini_api_key, commandr_api_key):
         self.prompt_dir = prompt_dir
         self.model = model
-        self.openai_api_key = openai_api_key    
-        self.claude_api_key = claude_api_key
-        self.gemini_api_key = gemini_api_key
-        self.commandr_api_key = commandr_api_key
+        self.openai_api_key = os.environ.get('OPENAI_API_KEY')      
+        self.claude_api_key = os.environ.get('ANTHROPIC_API_KEY')
+        self.gemini_api_key = os.environ.get('GEMINI_API_KEY')
+        self.commandr_api_key = os.environ.get('COMMANDR_API_KEY')
 
     def count_tokens(self, text):
         return len(self.encoding.encode(text))

utils.py

@@ -37,6 +37,9 @@ def process_paper(pdf_file, paper_dir, prompt_dir, api_keys):
     models = ['gpt-4-turbo-2024-04-09', 'gpt-4o', 'claude-3-opus-20240229', 'gemini-pro', 'command-r-plus']
     selected_models = random.sample(models, 2)
 
+    # REPLACE ONE OF THE MODELS WITH command-r-plus
+    # selected_models = ['gpt-4o', 'command-r-plus']
+
     reviews = []
 
     def process_with_model(model):