Update diarization_dataset.py

[BNarayanaReddy]

A robust way to find the duration of audio file instead of solely depending on the user for precised duration of the audio file.