Update script.py

script.py

@@ -156,6 +156,13 @@ with torch.inference_mode():
     )
 
 outputs = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0].strip()
-print("Output Type:", type(outputs))
+
+if "dataset1" in image_file:
+    print("Num of words: ", len(outputs))
+elif "dataset2" in image_file:
+    print()
+else:
+    print("Is single word?", len((outputs).split()) == 1)
+
 print(outputs)
 # End Llava inference