Update app.py

app.py

@@ -24,7 +24,11 @@ models=[
     "psyne/AIResnetClone",
 
 ]
-    
+
+fin_sum=[]
+#fin_res={f'{uid}':''}
+#fin_sum.append(fin_res)
+#tmp_res=
 def aiornot0(image):    
     labels = ["Real", "AI"]
     mod=models[0]
@@ -48,6 +52,8 @@ def aiornot0(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
+    tmp_res={f'{uid}-0':results}
+    fin_sum.append(tmp_res)
     return gr.HTML.update(html_out),results
 def aiornot1(image):    
     labels = ["Real", "AI"]
@@ -72,6 +78,8 @@ def aiornot1(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
+    tmp_res={f'{uid}-1':results}
+    fin_sum.append(tmp_res)        
     return gr.HTML.update(html_out),results    
 def aiornot2(image):    
     labels = ["AI", "Real"]
@@ -96,7 +104,8 @@ def aiornot2(image):
     results = {}
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
-    #results[labels['label']] = result['score']
+    tmp_res={f'{uid}-2':results}
+    fin_sum.append(tmp_res)
     return gr.HTML.update(html_out),results
 def aiornot3(image):    
     labels = ["Real", "AI"]
@@ -182,6 +191,10 @@ def load_url(url):
         image=None
         mes=f"Image not Found<br>Error: {e}"
     return image,mes
+
+def tot_prob():
+    print (fin_sum)
+    
 with gr.Blocks() as app:
     with gr.Row():
         with gr.Column():
@@ -219,9 +232,9 @@ with gr.Blocks() as app:
                 n_out5=gr.Label(label="Output")
                 outp5 = gr.HTML("""""")                    
     load_btn.click(load_url,in_url,[inp,mes])
-    btn.click(aiornot0,[inp],[outp0,n_out0])
-    btn.click(aiornot1,[inp],[outp1,n_out1])
-    btn.click(aiornot2,[inp],[outp2,n_out2])
+    btn.click(aiornot0,[inp],[outp0,n_out0]).then(fin_sum,None,None)
+    btn.click(aiornot1,[inp],[outp1,n_out1]).then(fin_sum,None,None)
+    btn.click(aiornot2,[inp],[outp2,n_out2]).then(fin_sum,None,None)
     #btn.click(aiornot3,[inp],[outp3,n_out3])
     #btn.click(aiornot4,[inp],[outp4,n_out4])
     #btn.click(aiornot5,[inp],[outp5,n_out5])