Update app.py

app.py

@@ -59,7 +59,11 @@ def aiornot0(image):
     Probabilites:<br>
     Real: {px[0][0]}<br>
     AI: {px[1][0]}"""
-    return gr.HTML.update(html_out)
+    results = {}
+    for idx,result in enumerate(px):
+        results[labels[idx]] = pdx[idx][0]
+    #results[labels['label']] = result['score']
+    return gr.HTML.update(html_out),results
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]
@@ -120,10 +124,10 @@ with gr.Blocks() as app:
                 lab2 = gr.HTML(f"""Testing on Model: {models[2]}""")
                 outp2 = gr.HTML("""""")            
                 n_out2=gr.Label()
-    btn.click(aiornot0,[inp],outp0)
+    btn.click(aiornot0,[inp],[outp0,n_out0])
     btn.click(aiornot1,[inp],outp1)
     btn.click(aiornot2,[inp],outp2)
-    btn.click(image_classifier0,[inp],n_out0)
+    #btn.click(image_classifier0,[inp],n_out0)
     btn.click(image_classifier1,[inp],n_out1)
     btn.click(image_classifier2,[inp],n_out2)    
 app.launch()