Fix ViT-5 model prediction output and logging

app.py

@@ -201,9 +201,9 @@ def predict_image(img, confidence_threshold):
     
     try:
         prediction_5 = clf_5(img_pilvits)
-        result_5 = {pred['label']: pred['score'] for pred in prediction_2}
+        result_5 = {pred['label']: pred['score'] for pred in prediction_5}
         result_5output = [5, 'ViT-base Newcomer', result_5['Realism'], result_5['Deepfake']]
-        print(result_2output)
+        print(result_5output)
         # Ensure the result dictionary contains all class names
         for class_name in class_names_5:
             if class_name not in result_5: