Update caption_generator_ds.py

caption_generator_ds.py

@@ -254,7 +254,7 @@ def main():
     print(f"Loading dataset: {args.dataset_name}")
     dataset = load_dataset(args.dataset_name)
     len_ = len(dataset["train"])
-    len_ = 10
+    #len_ = 10
     
     # Initialize a list to store captions
     captions = []