harshitface2003 commited on
Commit
0cbc129
·
verified ·
1 Parent(s): db9c4a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,7 +11,7 @@ import numpy as np
11
  import cv2
12
  from PIL import Image
13
 
14
- MODEL_PATH = r"pose_landmarker_lite.task"
15
 
16
  # Drawing landmarks
17
  # def draw_landmarks_on_image(rgb_image, detection_result):
@@ -41,7 +41,7 @@ MODEL_PATH = r"pose_landmarker_lite.task"
41
 
42
  def detection(image, conf_threshold=0.3):
43
  frame = cv2.flip(image, 1)
44
- rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
45
 
46
  rgb_frame=cv2.circle(rgb_frame,(90,90),(255,0,0),20)
47
  # mp_image = mp.Image(image_format=mp.ImageFormat.SRGB, data=rgb_frame)
@@ -56,7 +56,7 @@ def detection(image, conf_threshold=0.3):
56
 
57
 
58
  with gr.Blocks() as demo:
59
- image = WebRTC(label="Stream", mode="send-receive", modality="video", height=480, width=640, mirror_webcam=True)
60
  conf_threshold = gr.Slider(
61
  label="Confidence Threshold",
62
  minimum=0.0,
 
11
  import cv2
12
  from PIL import Image
13
 
14
+ # MODEL_PATH = r"pose_landmarker_lite.task"
15
 
16
  # Drawing landmarks
17
  # def draw_landmarks_on_image(rgb_image, detection_result):
 
41
 
42
  def detection(image, conf_threshold=0.3):
43
  frame = cv2.flip(image, 1)
44
+ # rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
45
 
46
  rgb_frame=cv2.circle(rgb_frame,(90,90),(255,0,0),20)
47
  # mp_image = mp.Image(image_format=mp.ImageFormat.SRGB, data=rgb_frame)
 
56
 
57
 
58
  with gr.Blocks() as demo:
59
+ image = WebRTC(label="Stream", mode="send-receive", modality="video", height=480, width=640, time_limit=10)
60
  conf_threshold = gr.Slider(
61
  label="Confidence Threshold",
62
  minimum=0.0,