Update app.py

app.py

@@ -16,11 +16,11 @@ TITLE = "# [Florence-2 SD3 Long Captioner](https://huggingface.co/gokaygokay/Flo
 
 def modify_caption(caption: str) -> str:
     """
-    Removes specific prefixes from captions.
+    Removes specific prefixes from captions if present, otherwise returns the original caption.
     Args:
         caption (str): A string containing a caption.
     Returns:
-        str: The caption with the prefix removed if it was present.
+        str: The caption with the prefix removed if it was present, or the original caption.
     """
     # Define the prefixes to remove
     prefix_substrings = [
@@ -37,7 +37,10 @@ def modify_caption(caption: str) -> str:
         return replacers[match.group(0)]
     
     # Apply the regex to the caption
-    return re.sub(pattern, replace_fn, caption, count=1, flags=re.IGNORECASE)
+    modified_caption = re.sub(pattern, replace_fn, caption, count=1, flags=re.IGNORECASE)
+    
+    # If the caption was modified, return the modified version; otherwise, return the original
+    return modified_caption if modified_caption != caption else caption
     
 @spaces.GPU
 def run_example(image):