Face_recognition

Sleeping

Kaushik066 commited on Feb 7

Commit

0d6332b

verified ·

1 Parent(s): be45b57

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -166,7 +166,7 @@ class CreateDatasetProd():
     def create_dataset(self, image_paths, webcam=False):
         if webcam == True:
-            pixel_values = torch.stack(self.get_pixel(image_paths))
         else:
             pixel_values = torch.stack(self.get_pixels(image_paths))

     def create_dataset(self, image_paths, webcam=False):
         if webcam == True:
+            pixel_values = self.get_pixel(image_paths)
         else:
             pixel_values = torch.stack(self.get_pixels(image_paths))