readability

2 years ago · 73ab648bbc
1 changed files with 1 additions and 1 deletions
--- a/matrix_pygmalion_bot/ai/runpod.py
+++ b/matrix_pygmalion_bot/ai/runpod.py
@ -115,7 +115,7 @@ async def generate_sync(
                    output.sort(key=len, reverse=True)
                    text = output[0]
                else:
-                    text = r_json["output"]
+                    text = output
                if api_mode == "runpod":
                    reply = text.removeprefix(prompt)
                else: