more fixes

app.py

@@ -206,13 +206,7 @@ def predict_image(img, confidence_threshold):
     except Exception as e:
         label_5 = f"Error: {str(e)}"
     
-        # Generate augmented images
-    augmented_img_flip, augmented_img_rotate = augment_image(img_pil)
-    
-    # Convert augmented images to bytes for display
-    augmented_img_flip_bytes = convert_pil_to_bytes(augmented_img_flip)
-    augmented_img_rotate_bytes = convert_pil_to_bytes(augmented_img_rotate)
-    
+
     # Combine results
     combined_results = {
         "SwinV2/detect": label_1,
@@ -224,9 +218,9 @@ def predict_image(img, confidence_threshold):
         # Generate HTML content
     
     combined_outputs = [ result_1output, result_2output, result_3output, result_4output, result_5output ]
-    html_content = generate_results_html(combined_outputs)
+    # html_content = generate_results_html(combined_outputs)
 
-    return img_pil, combined_outputs, html_content, augmented_img_flip_bytes, augmented_img_rotate_bytes
+    return img_pil, combined_outputs
 
 # Define a function to generate the HTML content
 # Define a function to generate the HTML content