Fix ViT-5 model prediction logic and output formatting

- Modify confidence threshold check to use 'Deepfake' instead of 'AI Image'
- Remove unnecessary print statement before final output print
- Adjust print statement placement for better logging
- Minor UI text spacing adjustment in results HTML template

app.py

@@ -203,13 +203,13 @@ def predict_image(img, confidence_threshold):
         prediction_5 = clf_5(img_pilvits)
         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_5output)
+        
         # Ensure the result dictionary contains all class names
         for class_name in class_names_5:
             if class_name not in result_5:
                 result_5[class_name] = 0.0
         # Check if either class meets the confidence threshold
-        if result_5['AI Image'] >= confidence_threshold:
+        if result_5['Deepfake'] >= confidence_threshold:
             label_5 = f"AI, Confidence: {result_5['Deepfake']:.4f}"
             result_5output += ['AI']
         elif result_5['Real Image'] >= confidence_threshold:
@@ -220,6 +220,8 @@ def predict_image(img, confidence_threshold):
             result_5output += ['UNCERTAIN']
     except Exception as e:
         label_5 = f"Error: {str(e)}"
+        
+    print(result_5output)
     
     
     # try:
@@ -452,7 +454,7 @@ def generate_results_html(results):
                 class="flex flex-col bg-gray-800 rounded-sm p-4 m-1 border border-gray-800 shadow-xs transition hover:shadow-lg dark:shadow-gray-700/25">
                 <div
                     class="-m-4 h-24 {get_header_color(results[4][-1])[0]} rounded-sm rounded-b-none transition border group-hover:border-gray-100 group-hover:shadow-lg group-hover:{get_header_color(results[4][-1])[4]}">
-                    <span class="text-gray-300 font-mono tracking-widest p-4 pb-3 block text-xs text-center">MODEL 5: </span>
+                    <span class="text-gray-300 font-mono tracking-widest p-4 pb-3 block text-xs text-center">MODEL 5:</span>
                     <span
                         class="flex w-24 mx-auto tracking-wide items-center justify-center rounded-full {get_header_color(results[4][-1])[2]} px-1 py-0.5 {get_header_color(results[4][-1])[3]}"
                     >