Clean caption if truncated

app.py

@@ -11,8 +11,17 @@ def get_caption(image_in):
 	    True,	# bool  in 'Enable detailed captioning' Checkbox component
 		fn_index=2
     )
-    print(f"IMAGE CAPTION: {fuyu_result}")
-    return fuyu_result
+
+    # Find the last occurrence of "."
+    last_period_index = fuyu_result.rfind('.')
+
+    # Truncate the string up to the last period
+    truncated_caption = fuyu_result[:last_period_index + 1]
+
+    # print(truncated_caption)
+    print(f"IMAGE CAPTION: {truncated_caption}")
+    
+    return truncated_caption
 
 import re
 import torch