Update app.py

app.py

@@ -44,12 +44,15 @@ def fn_video(video_file, conf_thres, iou_thres, start_sec, duration):
             except Exception as e:
                 print(e)
                 continue
-            print("FRAME DTYPE", type(frame))
-            out.write(pipeline(frame, conf_thres, iou_thres))
-            num_frames += 1
-            print("Processed {} frames".format(num_frames))
-            if num_frames == max_frames:
-                break
+            if (num_frames%10 != 0):
+                continue
+            else:
+                print("FRAME DTYPE", type(frame))
+                out.write(pipeline(frame, conf_thres, iou_thres))
+                num_frames += 1
+                print("Processed {} frames".format(num_frames))
+                if num_frames == max_frames:
+                    break
 
         out.release()
 
@@ -60,6 +63,11 @@ def fn_video(video_file, conf_thres, iou_thres, start_sec, duration):
     return out_file.name
 
 
+torch.hub.download_url_to_file('https://tochkanews.ru/wp-content/uploads/2020/09/0.jpg', '1.jpg')
+torch.hub.download_url_to_file('https://s.rdrom.ru/1/pubs/4/35893/1906770.jpg', '2.jpg')
+torch.hub.download_url_to_file('https://static.mk.ru/upload/entities/2022/04/17/07/articles/detailPicture/5b/39/28/b6/ffb1aa636dd62c30e6ff670f84474f75.jpg', '3.jpg')
+
+
 image_interface = gr.Interface(
     fn=fn_image,
     inputs=[
@@ -68,7 +76,7 @@ image_interface = gr.Interface(
         gr.Slider(0, 1, value=0.5, label="IOU Threshold"),
     ],
     outputs=gr.Image(type="file"),
-    examples=[["example_1.jpg", 0.5, 0.5], ["example_2.jpg", 0.25, 0.45], ["example_3.jpg", 0.25, 0.45]],
+    examples=[["1.jpg", 0.5, 0.5], ["2.jpg", 0.25, 0.45], ["3.jpg", 0.25, 0.45]],
     title="YOLOv6",
     description=(
         "Gradio demo for YOLOv6 for object detection on images. To use it, simply upload your image or click one of the"
@@ -90,7 +98,7 @@ video_interface = gr.Interface(
         gr.Slider(0, 1, value=0.25, label="Confidence Threshold"),
         gr.Slider(0, 1, value=0.45, label="IOU Threshold"),
         gr.Slider(0, 10, value=0, label="Start Second", step=1),
-        gr.Slider(0, 10 if pipeline.device.type != 'cpu' else 3, value=4, label="Duration", step=1),
+        gr.Slider(0, 10 if pipeline.device.type != 'cpu' else 3, value=3, label="Duration", step=1),
     ],
     outputs=gr.Video(type="file", format="mp4"),
     examples=[