Update app.py
Browse files
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)
|