BSuruchi commited on
Commit
5be6680
·
verified ·
1 Parent(s): 38c3202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -230,8 +230,9 @@ def run(
230
  mp_pose.POSE_CONNECTIONS,
231
  landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
232
  )
233
-
234
- return res[:, :, ::-1]
 
235
 
236
 
237
  model_complexities = list(range(3))
@@ -250,7 +251,7 @@ demo = gr.Interface(
250
  gr.Radio(label="Background Color", choices=background_colors, type="value", value=background_colors[0]),
251
  ],
252
 
253
- outputs=[gr.Image(label="Output"),gr.Textbox(label="Output Text")],
254
  examples=examples,
255
  title=TITLE,
256
  description=DESCRIPTION,
 
230
  mp_pose.POSE_CONNECTIONS,
231
  landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(),
232
  )
233
+ res, pose_classification = classify_pose(results.pose_landmarks.landmark, res) #Pose Classification code
234
+
235
+ return res[:, :, ::-1]
236
 
237
 
238
  model_complexities = list(range(3))
 
251
  gr.Radio(label="Background Color", choices=background_colors, type="value", value=background_colors[0]),
252
  ],
253
 
254
+ outputs=[gr.Image(label="Output"),
255
  examples=examples,
256
  title=TITLE,
257
  description=DESCRIPTION,