Update app.py

app.py

@@ -24,8 +24,12 @@ def image_classifier0(image):
     labels = ["Real", "AI"]
     outputs = pipe0(image)
     results = {}
+    result_test={}
     for idx,result in enumerate(outputs):
-        results[labels[idx]] = outputs[idx]['score']
+        result_test[labels[idx]] = outputs[idx]['score']
+    print (result_test)
+    for result in outputs:
+        results[result['label']] = result['score']    
     return results
 def image_classifier1(image):
     outputs = pipe1(image)