Update app.py

app.py

@@ -189,9 +189,9 @@ dataset_prod_obj = CreateDatasetProd(image_processor_prod)
 prod_ds = dataset_prod_obj.create_dataset(image_paths, webcam=False)
 prod_dl = DataLoader(prod_ds, batch_size=BATCH_SIZE)
 
-# Testing the dataloader
-prod_inputs = next(iter(prod_dl))
-st.write(prod_inputs['pixel_values'].shape) 
+## Testing the dataloader
+#prod_inputs = next(iter(prod_dl))
+#st.write(prod_inputs['pixel_values'].shape) 
 
 # Read image from Camera
 enable = st.checkbox("Enable camera")