Update app.py

app.py

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