Update app.py

app.py

@@ -107,8 +107,9 @@ class Pipeline:
 
     def __call__(self, video_path, temperature=0.7, top_p=0.95, max_text_length=77, num_return_sequences=10):
         text = ""
+        MAX_ITERATIONS = 5
         with torch.autocast(self.device):
-            for start, stop, frames in iter_clips(video_path):
+            for i, start, stop, frames in enumerate(iter_clips(video_path)):
                 text_to_add = f"{'-'*30} Predictions From: {start:2.3f}-{stop:2.3f} seconds {'-'*30}\n"
                 print(text_to_add)
                 text += text_to_add
@@ -134,6 +135,9 @@ class Pipeline:
                     text_to_add = '\t{}: {}\n'.format(i, generated_text_str)
                     print(text_to_add)
                     text += text_to_add
+
+                if (i+1) >= MAX_ITERATIONS:
+                    return text
         return text
 
 interface = gr.Interface(
@@ -143,7 +147,8 @@ interface = gr.Interface(
         gr.Slider(0.0, 1.0, 0.7, label='temperature'),
         gr.Slider(0.0, 1.0, 0.95, label='top_p'),
     ],
-    outputs='text'
+    outputs='text',
+    examples=[['eating_spaghetti.mp4', 0.7, 0.95], ['assets/3c0dffd0-e38e-4643-bc48-d513943dc20b_012_014.mp4', 0.7, 0.95]]
 )
 
 if __name__ == '__main__':