Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import json
+import re
 from gradio_client import Client
 
 #fusecap_client = Client("https://noamrot-fusecap-image-captioning.hf.space/")
@@ -31,7 +32,17 @@ def get_caption(image_in):
         reconstructed_sentence.append(sublist[0])
 
     full_sentence = ' '.join(reconstructed_sentence)
-    print(full_sentence)
+    #print(full_sentence)
+
+    # Find the pattern matching the expected format ("Describe this image in detail:" followed by optional space and then the rest)...
+    pattern = r'(?<=Describe this image in detail:\s*)(.*)$'
+    # Apply the regex pattern to extract the description text.
+    match = re.search(pattern, full_sentence)
+    if match:
+        description = match.group(1)
+        print(description)
+    else:
+        print("Unable to locate valid description.")
 
     # Find the last occurrence of "."
     #last_period_index = full_sentence.rfind('.')
@@ -42,7 +53,7 @@ def get_caption(image_in):
     # print(truncated_caption)
     #print(f"\n—\nIMAGE CAPTION: {truncated_caption}")
     
-    return full_sentence
+    return description
 
 import re
 import torch