updates

models/base.py

@@ -47,5 +47,5 @@ class BaseModel:
         predictions = resp.json()["predictions"]
         if isinstance(predictions[0], str):
             return predictions[0].strip()
-        predictions = sorted(predictions, key=lambda d: d['score'])
+        predictions = sorted(predictions[0], key=lambda d: d['score'])
         return predictions[-1]["text"].strip()