Update app.py

app.py

@@ -19,25 +19,10 @@ def get_caption_from_kosmos(image_in):
 
     print(f"KOSMOS2 RETURNS: {kosmos2_result}")
 
-    with open(kosmos2_result[1], 'r') as f:
-        data = json.load(f)
-    
-    reconstructed_sentence = []
-    for sublist in data:
-        reconstructed_sentence.append(sublist[0])
-
-    full_sentence = ' '.join(reconstructed_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.")
+    data = kosmos2_result[1]
+
+    # Extract and combine tokens starting from the second element
+    sentence = ''.join(item['token'] for item in data[1:])
 
     # Find the last occurrence of "."
     #last_period_index = full_sentence.rfind('.')
@@ -48,7 +33,7 @@ def get_caption_from_kosmos(image_in):
     # print(truncated_caption)
     #print(f"\n—\nIMAGE CAPTION: {truncated_caption}")
     
-    return description
+    return sentence
 
 def get_caption_from_MD(image_in):
     client = Client("https://vikhyatk-moondream1.hf.space/")