Update app.py

app.py

@@ -21,7 +21,7 @@ pipe1 = pipeline("image-classification", f"{models[1]}")
 pipe2 = pipeline("image-classification", f"{models[2]}")
 
 def image_classifier0(image):
-    labels = ["Real", "AI"]
+    labels = ["AI", "Real"]
     outputs = pipe0(image)
     results = {}
     result_test={}
@@ -32,14 +32,24 @@ def image_classifier0(image):
         results[result['label']] = result['score']    
     return results
 def image_classifier1(image):
+    labels = ["AI", "Real"]
     outputs = pipe1(image)
     results = {}
+    result_test={}
+    for idx,result in enumerate(outputs):
+        result_test[labels[idx]] = outputs[idx]['score']
+    print (result_test)    
     for result in outputs:
         results[result['label']] = result['score']
     return results
 def image_classifier2(image):
+    labels = ["AI", "Real"]
     outputs = pipe2(image)
     results = {}
+    result_test={}
+    for idx,result in enumerate(outputs):
+        result_test[labels[idx]] = outputs[idx]['score']
+    print (result_test)    
     for result in outputs:
         results[result['label']] = result['score']
     return results