Update app.py

app.py

@@ -14,6 +14,32 @@ from model import create_effnetb2_model
 from timeit import default_timer as timer
 from typing import Tuple, Dict
 
+file_urls = [
+    'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
+]
+
+def download_file(url, save_name):
+    url = url
+    if not os.path.exists(save_name):
+        file = requests.get(url)
+        open(save_name, 'wb').write(file.content)
+
+for i, url in enumerate(file_urls):
+    if 'mp4' in file_urls[i]:
+        download_file(
+            file_urls[i],
+            f"video.mp4"
+        )
+    else:
+        download_file(
+            file_urls[i],
+            f"image_{i}.jpg"
+        )
+
+
+video_path = [['video.mp4']]
+
+
 # Setup class names
 class_names = ["dented","good"]