Update src/litbench_pipeline.py

src/litbench_pipeline.py

@@ -64,7 +64,7 @@ def update_button_styles(selected_task):
     """Update button styles based on selection."""
     return [gr.update(variant="primary" if selected_task == prompt else "secondary") for prompt in task_list]
 
-
+@spaces.CPU
 # Fetch and store arXiv source files
 def fetch_arxiv_papers(papers_to_download):
     # Download the arXiv metadata file if it doesn't exist
@@ -311,7 +311,7 @@ def fetch_arxiv_papers(papers_to_download):
 
 
 # Chat prediction function
-@spaces.GPU
+@spaces.GPU(duration=120)
 def predict(message, history, selected_task):
     global model
     # Initialize the conversation string