Hurum Maksora Tohfa commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -284,7 +284,7 @@ if run_btn:
|
|
| 284 |
triager = PaperTriager(
|
| 285 |
primary_client, primary_cfg.model_id, primary_cfg.prices, tracker,
|
| 286 |
max_tokens=primary_cfg.triage_max_tokens,
|
| 287 |
-
batch_size=primary_cfg.
|
| 288 |
)
|
| 289 |
triaged = triager.triage(all_papers, spec)
|
| 290 |
top = triaged[:num_papers]
|
|
|
|
| 284 |
triager = PaperTriager(
|
| 285 |
primary_client, primary_cfg.model_id, primary_cfg.prices, tracker,
|
| 286 |
max_tokens=primary_cfg.triage_max_tokens,
|
| 287 |
+
batch_size=primary_cfg.score_batch_size,
|
| 288 |
)
|
| 289 |
triaged = triager.triage(all_papers, spec)
|
| 290 |
top = triaged[:num_papers]
|