Update app.py

app.py

@@ -53,10 +53,18 @@ def get_meta_from_video(Seg_Tracker, input_video, scale_slider, checkpoint):
     cap.release()
     output_frames = int(total_frames * scale_slider)
     frame_interval = max(1, total_frames // output_frames)
-    ffmpeg.input(input_video, hwaccel='cuda').output(
-        os.path.join(output_dir, '%07d.jpg'), q=2, start_number=0, 
-        vf=rf'select=not(mod(n\,{frame_interval}))', vsync='vfr'
-    ).run()
+    print(f"frame_interval: {frame_interval}")
+    try:
+        ffmpeg.input(input_video, hwaccel='cuda').output(
+            os.path.join(output_dir, '%07d.jpg'), q=2, start_number=0, 
+            vf=rf'select=not(mod(n\,{frame_interval}))', fps_mode='vfr'
+        ).run()
+    except:
+        print(f"ffmpeg cuda err")
+        ffmpeg.input(input_video).output(
+            os.path.join(output_dir, '%07d.jpg'), q=2, start_number=0, 
+            vf=rf'select=not(mod(n\,{frame_interval}))', fps_mode='vfr'
+        ).run()
 
     first_frame_path = os.path.join(output_dir, '0000000.jpg')
     first_frame = cv2.imread(first_frame_path)
@@ -368,6 +376,9 @@ def seg_track_app():
                 <a href="https://drive.google.com/drive/folders/1EXzRkxZmrXbahCFA8_ImFRM6wQDEpOSe?usp=sharing">
                     <img src="https://img.shields.io/badge/Video-Tutorial-green?style=plastic" alt="Video Tutorial" style="display:inline-block; margin-right:10px;">
                 </a>
+                <a href="https://github.com/bowang-lab/MedSAM/tree/MedSAM2?tab=readme-ov-file#fine-tune-sam2-on-the-abdomen-ct-dataset">
+                    <img src="https://img.shields.io/badge/Fine--tune-SAM2-blue" alt="Fine-tune SAM2" style="display:inline-block; margin-right:10px;">
+                </a>
             </div>
             <div style="text-align:left; margin-bottom:20px;">
                 This API supports using box (generated by scribble) and point prompts for video segmentation with 
@@ -386,6 +397,11 @@ def seg_track_app():
                     <li>8. Download the video with segmentation results</li>
                 </ol>
             </div>
+            <div style="text-align:left; margin-bottom:20px;">
+                Other useful resources: 
+                <a href="https://ai.meta.com/sam2" target="_blank">Official demo</a> from MetaAI, 
+                <a href="https://www.youtube.com/watch?v=Dv003fTyO-Y" target="_blank">Video tutorial</a> from Piotr Skalski.
+            </div>
             '''
         )
 
@@ -588,4 +604,4 @@ def seg_track_app():
     app.launch(debug=True, enable_queue=True, share=False)
 
 if __name__ == "__main__":
-    seg_track_app()
+    seg_track_app()