Update app.py

app.py

@@ -159,7 +159,7 @@ class CreateDatasetProd():
         # Scaling the video to ML model's desired format
         img = self.image_processor(img, return_tensors='pt') #, input_data_format='channels_first')
 
-        pixel_values = img['pixel_values'].squeeze(0)
+        pixel_values = img['pixel_values'] #.squeeze(0)
 
         # Force garbage collection
         del img