Update app.py

app.py

@@ -67,7 +67,8 @@ def predict_image(img, confidence_threshold):
     try:
         prediction_1 = clf_1(img_pil)
         result_1 = {pred['label']: pred['score'] for pred in prediction_1}
-        print(result_1)
+        result_1output = [1, result_1['artificial'], result_1['real']]
+        print(result_1output)
         # Ensure the result dictionary contains all class names
         for class_name in class_names_1:
             if class_name not in result_1:
@@ -109,8 +110,8 @@ def predict_image(img, confidence_threshold):
             logits_3 = outputs_3.logits
             probabilities_3 = softmax(logits_3.cpu().numpy()[0])
         result_3 = {
-            labels_3[0]: float(probabilities_3[0]),  # AI
             labels_3[1]: float(probabilities_3[1])   # Real
+            labels_3[0]: float(probabilities_3[0]),  # AI
         }
         print(result_3)
         # Ensure the result dictionary contains all class names
@@ -135,8 +136,8 @@ def predict_image(img, confidence_threshold):
             logits_4 = outputs_4.logits
             probabilities_4 = softmax(logits_4.cpu().numpy()[0])
         result_4 = {
-            labels_4[0]: float(probabilities_4[0]),  # AI
             labels_4[1]: float(probabilities_4[1])   # Real
+            labels_4[0]: float(probabilities_4[0]),  # AI
         }
         print(result_4)
         # Ensure the result dictionary contains all class names