jsonify text response string

app.py

@@ -4,6 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import spaces
 import openai
+import json
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 LEPTON_API_TOKEN = os.environ.get("LEPTON_API_TOKEN", None)
@@ -96,7 +97,7 @@ def model_call(question, document, answer):
         prompt=NEW_FORMAT
     )
     print("RESPONSE FROM CLIENT:", response)
-    generated_text = response.choices[0].text
+    generated_text = json(response.choices[0].text)
     reasoning = generated_text["REASONING"][0]
     score = generated_text["SCORE"]
     # inputs = tokenizer(NEW_FORMAT, return_tensors="pt")