updates

models/base.py

@@ -44,4 +44,8 @@ class BaseModel:
                            f"1. Scale-to-Zero enabled, so please wait for some minutes and try again. "
                            f"2. Probably the response generated by the model is to big, try changing max_new_tokens. "
                            f"3. If nothing helps — report the problem.")
-        return resp.json()["predictions"][0].strip()
+        predictions = resp.json()["predictions"]
+        if isinstance(predictions[0], str):
+            return predictions[0].strip()
+        predictions = sorted(predictions, key=lambda d: d['score'])
+        return predictions[-1]["text"].strip()