Update app.py

app.py

@@ -15,6 +15,30 @@ models=[
     "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]}")
+
+def image_classifier0(image):
+    outputs = pipe0(image)
+    results = {}
+    for result in outputs:
+        results[result['label']] = result['score']
+    return results
+def image_classifier1(image):
+    outputs = pipe1(image)
+    results = {}
+    for result in outputs:
+        results[result['label']] = result['score']
+    return results
+def image_classifier2(image):
+    outputs = pipe2(image)
+    results = {}
+    for result in outputs:
+        results[result['label']] = result['score']
+    return results
+
+    
 def aiornot0(image):    
     labels = ["Real", "AI"]
     mod=models[0]
@@ -87,14 +111,19 @@ with gr.Blocks() as app:
             with gr.Box():
                 lab0 = gr.HTML(f"""Testing on Model: {models[0]}""")
                 outp0 = gr.HTML("""""")
+                n_out0=gr.Label()
             with gr.Box():
                 lab1 = gr.HTML(f"""Testing on Model: {models[1]}""")
                 outp1 = gr.HTML("""""")
+                n_out1=gr.Label()
             with gr.Box():
                 lab2 = gr.HTML(f"""Testing on Model: {models[2]}""")
                 outp2 = gr.HTML("""""")            
+                n_out2=gr.Label()
     btn.click(aiornot0,[inp],outp0)
     btn.click(aiornot1,[inp],outp1)
     btn.click(aiornot2,[inp],outp2)
-    
+    btn.click(image_classifier0,[inp],n_out0)
+    btn.click(image_classifier1,[inp],n_out1)
+    btn.click(image_classifier2,[inp],n_out2)    
 app.launch()