From cdd3b08b85e2f2e0c68be7151e95fa4f00d5bc3b Mon Sep 17 00:00:00 2001
From: seranus <ognjen_j@yahoo.com>
Date: Mon, 6 Dec 2021 23:30:36 +0100
Subject: [PATCH] fix

---
 samplelib/SampleGeneratorFace.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/samplelib/SampleGeneratorFace.py b/samplelib/SampleGeneratorFace.py
index 70f8d71..68229c2 100644
--- a/samplelib/SampleGeneratorFace.py
+++ b/samplelib/SampleGeneratorFace.py
@@ -142,4 +142,6 @@ class SampleGeneratorFace(SampleGeneratorBase):
                 for i in range(len(x)):
                     batches[i].append ( x[i] )
 
+                filenames.append(sample.filename)
+
             yield ([ np.array(batch) for batch in batches], filenames)