update v2

src/api.py

@@ -47,7 +47,7 @@ def load_writing_model(task_name, model_id):
         print(f"Loading {task_name}: {model_id}...")
         pipelines[task_name] = pipeline(
             "text-classification", model=model_id, tokenizer=model_id,
-            device=DEVICE, return_all_scores=True
+            device=DEVICE, top_k=None
         )
         print(f"Loaded {task_name}.")
     except Exception as e: