Rohit1412 commited on
Commit
1f5db55
·
verified ·
1 Parent(s): 9e5613d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,8 +3,8 @@ import cv2
3
  import numpy as np
4
  import torch
5
 
6
- # Load the YOLOv7 model
7
- model = torch.hub.load('WongKinYiu/yolov7', 'yolov7')
8
 
9
  def detect_objects(image):
10
  img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
@@ -24,7 +24,6 @@ def detect_objects(image):
24
  return annotated_image
25
 
26
  def detect_live_objects(video):
27
- # Capture frames from the video feed and process them
28
  img = cv2.cvtColor(video, cv2.COLOR_RGB2BGR)
29
  results = model(img)
30
 
 
3
  import numpy as np
4
  import torch
5
 
6
+ # Load the YOLOv7 model with force_reload to ensure fresh download
7
+ model = torch.hub.load('WongKinYiu/yolov7', 'yolov7', force_reload=True)
8
 
9
  def detect_objects(image):
10
  img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
 
24
  return annotated_image
25
 
26
  def detect_live_objects(video):
 
27
  img = cv2.cvtColor(video, cv2.COLOR_RGB2BGR)
28
  results = model(img)
29