Fucius commited on
Commit
795b262
·
verified ·
1 Parent(s): c69ab70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -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)
 
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)