Separate output into two textboxes

app.py

@@ -97,6 +97,8 @@ def model_call(question, document, answer):
     )
     print("RESPONSE FROM CLIENT:", response)
     generated_text = response.choices[0].text
+    reasoning = generated_text["REASONING"][0]
+    score = generated_text["SCORE"]
     # inputs = tokenizer(NEW_FORMAT, return_tensors="pt")
     # print("INPUTS", inputs)
     # input_ids = inputs.input_ids
@@ -112,8 +114,8 @@ def model_call(question, document, answer):
     #     outputs = model.generate(**generate_kwargs)
     # print("OUTPUTS", outputs)
     # generated_text = tokenizer.decode(outputs[0])
-    print(generated_text)
-    return generated_text
+    # print(generated_text)
+    return reasoning, score
 
 # def update_model(model_choice, tokenizer_state, model_state):
 #     new_tokenizer, new_model = load_model_and_tokenizer(model_choice)
@@ -125,13 +127,16 @@ inputs = [
     gr.Textbox(label="Document"),
     gr.Textbox(label="Answer")
 ]
-# output = gr.Textbox(label="Output")
+outputs = [
+    gr.Textbox(label="Reasoning"),
+    gr.Textbox(label="Score")
+]
 
 # submit_button = gr.Button("Submit")
 
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
-    gr.Interface(fn=model_call, inputs=inputs, outputs="text")
+    gr.Interface(fn=model_call, inputs=inputs, outputs=outputs)
     # tokenizer_state = gr.State()
     # model_state = gr.State()