Update app.py

app.py

@@ -92,9 +92,9 @@ def submit(image, *agents):
 
     Answer = 'O'
     for k in range(TOTAL_K):
-        agents[k].update(value = Answer)  
+        agents[k] = Answer  
         
-    return np.array(new_image)
+    return np.array(new_image), agents
 
 theme = gr.themes.Default(primary_hue="green")
 agents = []
@@ -165,7 +165,7 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
     b0.click(random_sample, inputs = [], outputs = [image])
     b2.click(reset_sample, inputs = [], outputs = [image])
 
-    b1.click(submit, inputs = [image, *agents], outputs = [image])
+    b1.click(submit, inputs = [image, *agents], outputs = [image, *agents])
                   
     # inputs.submit(reset_textbox, [], [inputs, b1], queue=False)
     # inputs.submit(predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code, inputs, b1],)  #openai_api_key