Update app.py

app.py

@@ -27,6 +27,11 @@ import zipfile
 
 import pathlib
 
+class PyDataset(tf.keras.utils.data_utils.Dataset):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+  
+
 
 # Ensure that these imports are at the beginning of your script to avoid any NameError issues.