increase duration

app.py

@@ -13,10 +13,10 @@ from paladin import run_paladin
 
 NUM_WORKERS = int(os.getenv("NUM_WORKERS", 2))
 
-@spaces.GPU
+@spaces.GPU(duration=300)
 def get_features_and_infer(coords, slide_path, attrs, site_type):
     use_gpu = torch.cuda.is_available()
-    features, _ = get_features(coords, slide_path, attrs, model_type=ModelType.OPTIMUS, use_gpu=use_gpu, batch_size=64, num_workers=NUM_WORKERS)
+    features, _ = get_features(coords, slide_path, attrs, model_type=ModelType.OPTIMUS, use_gpu=use_gpu, batch_size=256, num_workers=NUM_WORKERS)
 
     # Step 3: Run Aeon to predict histology
     aeon_results, _ = run_aeon(