attempt: fix(bug)

app.py

@@ -116,7 +116,7 @@ def predict_image(img, confidence_threshold):
     label_4, result_4output = predict_with_model(img_pil, model_4, CLASS_NAMES["model_4"], confidence_threshold, "SDXL + FLUX", 4, feature_extractor_4)
     label_5, result_5output = predict_with_model(img_pilvits, clf_5, CLASS_NAMES["model_5"], confidence_threshold, "ViT-base Newcomer", 5)
     label_5b, result_5boutput = predict_with_model(img_pilvits, clf_5b, CLASS_NAMES["model_5b"], confidence_threshold, "ViT-base Newcomer", 6)
-    label_6, result_6output = predict_with_model(img_pilvits, model_6, CLASS_NAMES["model_6"], confidence_threshold, "Swin Midjourney/SDXL", 6, feature_extractor_6)
+    label_6, result_6output = predict_with_model(img_pilvits, model_6, CLASS_NAMES["model_6"], confidence_threshold, "Swin Midjourney/SDXL", 7, feature_extractor_6)
 
     combined_results = {
         "SwinV2/detect": label_1,
@@ -127,6 +127,7 @@ def predict_image(img, confidence_threshold):
         "prithivMLmods-2-22": label_5b,
         "SwinMidSDXL": label_6
     }
+    print(combined_results)
 
     combined_outputs = [result_1output, result_2output, result_3output, result_4output, result_5output, result_5boutput, result_6output]
     return img_pil, combined_outputs