Update app.py

app.py

@@ -21,7 +21,7 @@ models=[
     "arnolfokam/ai-generated-image-detector",
 
 ]
-   
+fin_sum=[]   
 def aiornot0(image):    
     labels = ["Real", "AI"]
     mod=models[0]
@@ -45,6 +45,7 @@ def aiornot0(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
+    fin_sum.append(results)
     return gr.HTML.update(html_out),results
 def aiornot1(image):    
     labels = ["Real", "AI"]
@@ -69,6 +70,7 @@ def aiornot1(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
+    fin_sum.append(results)
     return gr.HTML.update(html_out),results    
 def aiornot2(image):    
     labels = ["AI", "Real"]
@@ -94,6 +96,8 @@ def aiornot2(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
+    fin_sum.append(results)
+        
     return gr.HTML.update(html_out),results
 
 def load_url(url):
@@ -107,6 +111,24 @@ def load_url(url):
         image=None
         mes=f"Image not Found<br>Error: {e}"
     return image,mes
+
+def tot_prob():
+    try:
+        fin_out = fin_sum[0]["Real"]+fin_sum[1]["Real"]+fin_sum[2]["Real"]
+        fin_out = fin_out/3
+        fin_sub = 1-fin_out
+        out={
+            "Real":f"{fin_out}",
+            "AI":f"{fin_sub}"
+        }
+        fin_sum.clear()
+        print (fin_out)
+        return out
+    except Exception as e:
+        pass
+        print (e)
+
+    
 with gr.Blocks() as app:
     with gr.Column():
         in_url=gr.Textbox(label="Image URL")
@@ -119,7 +141,9 @@ with gr.Blocks() as app:
         with gr.Row():
             with gr.Box():
                 lab0 = gr.HTML(f"""<b>Testing on Model: <a href='https://huggingface.co/{models[0]}'>{models[0]}</a></b>""")
+            with gr.Box():
                 lab1 = gr.HTML(f"""<b>Testing on Model: <a href='https://huggingface.co/{models[1]}'>{models[1]}</a></b>""")
+            with gr.Box():
                 lab2 = gr.HTML(f"""<b>Testing on Model: <a href='https://huggingface.co/{models[2]}'>{models[2]}</a></b>""")
                 
         with gr.Row():
@@ -133,9 +157,11 @@ with gr.Blocks() as app:
                 n_out2=gr.Label(label="Output")
                 outp2 = gr.HTML("""""")    
           
+        with gr.Row():
+            fin=gr.Label(label="Final Probability")        
     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(tot_prob,None,fin)
+    btn.click(aiornot1,[inp],[outp1,n_out1]).then(tot_prob,None,fin)
+    btn.click(aiornot2,[inp],[outp2,n_out2]).then(tot_prob,None,fin)
     
 app.queue(concurrency_count=20).launch()