Update app.py

app.py

@@ -14,7 +14,7 @@ models=[
     "arnolfokam/ai-generated-image-detector",
     "umm-maybe/AI-image-detector",
 ]
-
+'''
 pipe0 = pipeline("image-classification", f"{models[0]}")
 pipe1 = pipeline("image-classification", f"{models[1]}")
 pipe2 = pipeline("image-classification", f"{models[2]}")
@@ -37,7 +37,7 @@ def image_classifier2(image):
     for result in outputs:
         results[result['label']] = result['score']
     return results
-
+'''
     
 def aiornot0(image):    
     labels = ["Real", "AI"]
@@ -84,7 +84,11 @@ def aiornot1(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]] = px[idx][0]
+    #results[labels['label']] = result['score']
+    return gr.HTML.update(html_out),results    
 def aiornot2(image):    
     labels = ["AI", "Real"]
     mod=models[2]
@@ -105,7 +109,13 @@ def aiornot2(image):
     Probabilites:<br>
     Real: {px[1][0]}<br>
     AI: {px[0][0]}"""
-    return gr.HTML.update(html_out)
+
+    results = {}
+    for idx,result in enumerate(px):
+        results[labels[idx]] = px[idx][0]
+    #results[labels['label']] = result['score']
+    return gr.HTML.update(html_out),results
+    
 with gr.Blocks() as app:
     with gr.Column():
         inp = gr.Pil()
@@ -125,9 +135,9 @@ with gr.Blocks() as app:
                 outp2 = gr.HTML("""""")            
                 n_out2=gr.Label(label="Output")
     btn.click(aiornot0,[inp],[outp0,n_out0])
-    btn.click(aiornot1,[inp],outp1)
-    btn.click(aiornot2,[inp],outp2)
+    btn.click(aiornot1,[inp],[outp1,n_out1])
+    btn.click(aiornot2,[inp],[outp2,n_out2])
     #btn.click(image_classifier0,[inp],n_out0)
-    btn.click(image_classifier1,[inp],n_out1)
-    btn.click(image_classifier2,[inp],n_out2)    
+    #btn.click(image_classifier1,[inp],n_out1)
+    #btn.click(image_classifier2,[inp],n_out2)    
 app.launch()