update app.py # 4

app.py

@@ -61,26 +61,23 @@ def predict(input_file, downsample_size):
     # indices[np.asarray(rv_indices)] = 1
     # print("images selected")
 
-    indices = [0] * 1000
-    indices[0] = 1
-
-    global_ctr = 0
-    for fp_i, file_path in enumerate([input_file]):
-        for i, img in enumerate(video_reader(file_path,
-                                             targetFPS=9,
-                                             targetWidth=None,
-                                             to_rgb=False)):
-            if indices[global_ctr] == 1:
-                cv2.imwrite(join(selected_directory, str(global_ctr) + ".jpg"), img)
-            global_ctr += 1
-    print("selected images extracted")
-
-    all_selected_imgs_path = [join(selected_directory, f) for f in os.listdir(selected_directory) if
-                              isfile(join(selected_directory, f))]
+    # global_ctr = 0
+    # for fp_i, file_path in enumerate([input_file]):
+    #     for i, img in enumerate(video_reader(file_path,
+    #                                          targetFPS=9,
+    #                                          targetWidth=None,
+    #                                          to_rgb=False)):
+    #         if indices[global_ctr] == 1:
+    #             cv2.imwrite(join(selected_directory, str(global_ctr) + ".jpg"), img)
+    #         global_ctr += 1
+    # print("selected images extracted")
+    #
+    # all_selected_imgs_path = [join(selected_directory, f) for f in os.listdir(selected_directory) if
+    #                           isfile(join(selected_directory, f))]
 
     zipf = zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED)
-    for i, f in enumerate(all_selected_imgs_path):
-        zipf.write(f, basename(f))
+    # for i, f in enumerate(all_selected_imgs_path):
+    #     zipf.write(f, basename(f))
     zipf.close()
     print("selected images zipped")