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)
|