Update app.py

app.py

@@ -144,16 +144,18 @@ def preprocess_and_invert(input_video,
         preprocess_config['frames'], frames_per_second = video_to_frames(input_video)
         preprocess_config['data_path'] = input_video.split(".")[0]
 
-        total_vid_duration = len(preprocess_config['frames'])/frames_per_second
+        total_vid_frames = len(preprocess_config['frames'])
+        total_vid_duration = total_vid_frames/frames_per_second
         
         if(total_vid_duration < 1):
-            preprocess_config['n_frames'] = preprocess_config['frames']
+            preprocess_config['n_frames'] = total_vid_frames
         else:
             preprocess_config['n_frames'] = int(frames_per_second/n_seconds)
         
         if preprocess_config['n_frames'] % batch_size != 0:
             preprocess_config['batch_size'] = largest_divisor(batch_size)
-        
+
+        print("Running with batch size of ", preprocess_config['batch_size'])
         if randomize_seed:
             seed = randomize_seed_fn()
         seed_everything(seed)