Spaces:
Running
Running
Commit ·
2cdc5a5
1
Parent(s): f0f29fe
Update app.py
Browse files
app.py
CHANGED
|
@@ -69,10 +69,6 @@ def process_and_predict(url=None,
|
|
| 69 |
keypoint_radius=5
|
| 70 |
)
|
| 71 |
|
| 72 |
-
# Convert the resulting visualization to a PIL Image
|
| 73 |
-
# output_image_pil = Image.fromarray(output_image.astype('uint8'), 'RGB')
|
| 74 |
-
|
| 75 |
-
# Return the PIL Image directly
|
| 76 |
return output_image, skeleton_image
|
| 77 |
|
| 78 |
# Define the Gradio interface
|
|
@@ -89,12 +85,11 @@ iface = gr.Interface(
|
|
| 89 |
gr.components.Image(label="Skeleton Only")
|
| 90 |
],
|
| 91 |
title="YOLO-NAS-Pose Demo",
|
| 92 |
-
|
| 93 |
live=True,
|
| 94 |
allow_flagging=False,
|
| 95 |
|
| 96 |
)
|
| 97 |
|
| 98 |
# Launch the interface
|
| 99 |
-
iface.launch()
|
| 100 |
-
|
|
|
|
| 69 |
keypoint_radius=5
|
| 70 |
)
|
| 71 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
return output_image, skeleton_image
|
| 73 |
|
| 74 |
# Define the Gradio interface
|
|
|
|
| 85 |
gr.components.Image(label="Skeleton Only")
|
| 86 |
],
|
| 87 |
title="YOLO-NAS-Pose Demo",
|
| 88 |
+
description="Upload an image, enter an image URL, or use your webcam to use a pretrained YOLO-NAS-Pose L for inference. Get more hands-on with the [starter notebook for inference](https://bit.ly/yn-pose-inference), and learn how to fine-tune your own model with the [fine-tuning notebook](https://bit.ly/yn-pose-fine-tuning).",
|
| 89 |
live=True,
|
| 90 |
allow_flagging=False,
|
| 91 |
|
| 92 |
)
|
| 93 |
|
| 94 |
# Launch the interface
|
| 95 |
+
iface.launch()
|
|
|