Prathamesh1420 commited on
Commit
78b0fae
·
verified ·
1 Parent(s): 26663dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ if image is not None:
28
  cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
29
 
30
  # Perform fire detection
31
- results = model(cv2_img, device=device)
32
 
33
  # Draw bounding boxes for detected fires
34
  for result in results:
 
28
  cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
29
 
30
  # Perform fire detection
31
+ results = model(cv2_img)
32
 
33
  # Draw bounding boxes for detected fires
34
  for result in results: