Minor changes

app.py

@@ -61,8 +61,7 @@ def get_image_features(model, image_dir):
         dataset,
         batch_size=32,
         shuffle=False,
-        num_workers=8,
-        persistent_workers=True,
+        num_workers=4,
         drop_last=False,
     )