Milkigit
diff --git a/‎examples/pytorchvideo_tutorial.ipynb
+24-7 b/‎examples/pytorchvideo_tutorial.ipynb
+24-7
@@ -209,21 +209,38 @@
     "import subprocess\n",
     "\n",
     "import youtube_dl\n",
+    "from youtube_dl.utils import (DownloadError, ExtractorError)\n",
     "\n",
     "def download_video(url, start, dur, output):\n",
+    "    output_tmp = os.path.join(\"/tmp\",os.path.basename(output))\n",
     "    try:\n",
-    "        # From https://stackoverflow.com/questions/57131049/is-it-possible-to-download-a-specific-part-of-a-file\n",
+    "    # From https://stackoverflow.com/questions/57131049/is-it-possible-to-download-a-specific-part-of-a-file\n",
     "        with youtube_dl.YoutubeDL({'format': 'best'}) as ydl:\n",
     "            result = ydl.extract_info(url, download=False)\n",
     "            video = result['entries'][0] if 'entries' in result else result\n",
     "        \n",
     "        url = video['url']\n",
-    "        cmd = ['ffmpeg', '-i', url, '-ss', start, '-t', dur, '-c:v',\n",
-    "                'copy', '-c:a', 'copy', output]\n",
+    "        if start < 5:\n",
+    "            offset = start\n",
+    "        else:\n",
+    "            offset = 5\n",
+    "        start -= offset\n",
+    "        offset_dur = dur + offset\n",
+    "        start_str = str(timedelta(seconds=start)) \n",
+    "        dur_str = str(timedelta(seconds=offset_dur)) \n",
+    "\n",
+    "        cmd = ['ffmpeg', '-i', url, '-ss', start_str, '-t', dur_str, '-c:v',\n",
+    "                'copy', '-c:a', 'copy', output_tmp]\n",
+    "        subprocess.call(cmd)\n",
+    "\n",
+    "        start_str_2 = str(timedelta(seconds=offset)) \n",
+    "        dur_str_2 = str(timedelta(seconds=dur)) \n",
+    "\n",
+    "        cmd = ['ffmpeg', '-i', output_tmp, '-ss', start_str_2, '-t', dur_str_2, output]\n",
     "        subprocess.call(cmd)\n",
     "        return True\n",
     "        \n",
-    "    except:\n",
+    "    except (DownloadError, ExtractorError) as e:\n",
     "        print(\"Failed to download %s\" % output)\n",
     "        return False\n",
     "        \n",
@@ -252,13 +269,13 @@
     "        if not os.path.exists(c_dir):\n",
     "            os.makedirs(c_dir)\n",
     "        \n",
-    "        start_str = str(timedelta(seconds=segment[0])) \n",
-    "        dur_str = str(timedelta(seconds=dur)) \n",
+    "\n",
+    "        start = segment[0]\n",
     "        output = os.path.join(c_dir, \"%s_%s.mp4\" % (label.replace(\" \",\"_\"), fn))\n",
     "        \n",
     "        results = True\n",
     "        if not os.path.exists(output):\n",
-    "            result = download_video(url, start_str, dur_str, output)\n",
+    "            result = download_video(url, start, dur, output)\n",
     "        if result:\n",
     "            classes_count[label] += 1\n",
     "\n",