Update app.py

app.py

@@ -25,13 +25,19 @@ def get_caption(image_in):
 
     with open(kosmos2_result[1], 'r') as f:
         data = json.load(f)
-        print(data)
+    
+    reconstructed_sentence = []
+    for sublist in data:
+        reconstructed_sentence.append(sublist[0])
+
+    full_sentence = ' '.join(reconstructed_sentence)
+    print(full_sentence)
 
     # Find the last occurrence of "."
-    last_period_index = kosmos2_result[1].rfind('.')
+    last_period_index = full_sentence.rfind('.')
 
     # Truncate the string up to the last period
-    truncated_caption = kosmos2_result[1][:last_period_index + 1]
+    truncated_caption = full_sentence[:last_period_index + 1]
 
     # print(truncated_caption)
     print(f"\n—\nIMAGE CAPTION: {truncated_caption}")