Fucius commited on
Commit
73d1b00
1 Parent(s): 4b491e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,7 +28,7 @@ import cv2
28
  from PIL import Image, ImageOps
29
  from transformers import DPTFeatureExtractor, DPTForDepthEstimation
30
  # from controlnet_aux import OpenposeDetector
31
- from controlnet_aux.open_pose.body import Body
32
 
33
  try:
34
  from inference.models import YOLOWorld
@@ -328,8 +328,8 @@ def main(device, segment_type):
328
 
329
  depth_estimator = DPTForDepthEstimation.from_pretrained(args.dpt_checkpoint).to("cuda")
330
  feature_extractor = DPTFeatureExtractor.from_pretrained(args.dpt_checkpoint)
331
- body_model = Body(args.pose_detector_checkpoint)
332
- openpose = OpenposeDetector(body_model)
333
 
334
  def remove_tips():
335
  return gr.update(visible=False)
 
28
  from PIL import Image, ImageOps
29
  from transformers import DPTFeatureExtractor, DPTForDepthEstimation
30
  # from controlnet_aux import OpenposeDetector
31
+ # from controlnet_aux.open_pose.body import Body
32
 
33
  try:
34
  from inference.models import YOLOWorld
 
328
 
329
  depth_estimator = DPTForDepthEstimation.from_pretrained(args.dpt_checkpoint).to("cuda")
330
  feature_extractor = DPTFeatureExtractor.from_pretrained(args.dpt_checkpoint)
331
+ # body_model = Body(args.pose_detector_checkpoint)
332
+ # openpose = OpenposeDetector(body_model)
333
 
334
  def remove_tips():
335
  return gr.update(visible=False)