fix: string indices must be integers

src/api.py

@@ -210,8 +210,12 @@ async def grade_writing(request: WritingRequest):
     else:
         final_input = f"{request.prompt} [SEP] {request.essay}"
 
-    results = model(final_input, truncation=True, max_length=512)[0]
-    raw = {item['label']: item['score'] for item in results}
+    out = model(final_input, truncation=True, max_length=512)
+
+    # sometimes out = [[{label,score}...]]; sometimes out = [{label,score}...]
+    results = out[0] if (isinstance(out, list) and len(out) and isinstance(out[0], list)) else out
+
+    raw = {item["label"]: item["score"] for item in results}
     
     def r(x): return round(x * 2) / 2