Update app.py

app.py

@@ -65,6 +65,24 @@ def prepare_image(image, target_size):
 
     return np.expand_dims(image_array, axis=0)  # Add batch dimension
 
+# Function to process predictions with thresholds
+def process_predictions_with_thresholds(preds, tag_data, character_thresh, general_thresh, hide_rating_tags, character_tags_first):
+    # Extract prediction scores
+    scores = preds.flatten()
+
+    # Filter and sort character and general tags based on thresholds
+    character_tags = [tag_data.names[i] for i in tag_data.character if scores[i] >= character_thresh]
+    general_tags = [tag_data.names[i] for i in tag_data.general if scores[i] >= general_thresh]
+
+    # Optionally filter rating tags
+    rating_tags = [] if hide_rating_tags else [tag_data.names[i] for i in tag_data.rating]
+
+    # Sort tags based on user preference
+    final_tags = character_tags + general_tags if character_tags_first else general_tags + character_tags
+    final_tags += rating_tags  # Add rating tags at the end if not hidden
+
+    return final_tags
+
 class LabelData:
     def __init__(self, names, rating, general, character):
         self.names = names