Update app.py

app.py

@@ -231,9 +231,8 @@ class VideoClassifier:
         #         transcript=self.audio_extraction(video_input)
         #     except:
         #         transcript=self.audio_extraction_space(video_input)
-
+        start_time_caption = time.time()
         if checkbox=="Image Captions":
-            start_time_caption = time.time()
             video = cv2.VideoCapture(video_input)
             length = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
             no_of_frame = int(self.no_of_frames)
@@ -258,11 +257,12 @@ class VideoClassifier:
             print("CAPTIONS", captions)
             video.release()
             cv2.destroyAllWindows()
-            end_time_caption = time.time()
-            caption_time=end_time_caption-start_time_caption
+            
             # print("TIME TAKEN FOR IMAGE CAPTIONING", end_time_caption-start_time_caption)
         else:
-            caption=""
+            captions=""
+        end_time_caption = time.time()
+        caption_time=end_time_caption-start_time_caption
         start_time_generation = time.time()
         main_categories = Path("main_classes.txt").read_text()
         main_categories_list = ['Automotive', 'Books and Literature', 'Business and Finance', 'Careers', 'Education','Family and Relationships',