Update app.py

app.py

@@ -64,7 +64,7 @@ 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)
         # Ensure the result dictionary contains all class names
         for class_name in class_names_1:
             if class_name not in result_1:
@@ -84,7 +84,7 @@ def predict_image(img, confidence_threshold):
     try:
         prediction_2 = clf_2(img_pil)
         result_2 = {pred['label']: pred['score'] for pred in prediction_2}
-        
+        print(result_2)
         # Ensure the result dictionary contains all class names
         for class_name in class_names_2:
             if class_name not in result_2:
@@ -107,12 +107,12 @@ def predict_image(img, confidence_threshold):
             outputs_3 = model_3(**inputs_3)
             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
         }
-        
+        print(result_3)
         # Ensure the result dictionary contains all class names
         for class_name in labels_3:
             if class_name not in result_3:
@@ -140,7 +140,7 @@ def predict_image(img, confidence_threshold):
             labels_4[0]: float(probabilities_4[0]),  # AI
             labels_4[1]: float(probabilities_4[1])   # Real
         }
-        
+        print(result_4)
         # Ensure the result dictionary contains all class names
         for class_name in labels_4:
             if class_name not in result_4: