Spaces:

the-future-dev
/

chatBot-kosmos2

Runtime error

the-future-dev commited on Mar 4

Commit

a44470c

•

1 Parent(s): beccfdf

first commit

Files changed (1) hide show

kosmos.py CHANGED Viewed

@@ -23,4 +23,5 @@ def single_image_classification(image, prompt="", max_new_tokens=30):
     caption, entities = processor.post_process_generation(generated_text)
     print("ROBOT:", caption)
-    return processed_text

     caption, entities = processor.post_process_generation(generated_text)
     print("ROBOT:", caption)
+    print("ENTITIES: ", entities)
+    return caption