harpreetsahota commited on
Commit
2cdc5a5
β€’
1 Parent(s): f0f29fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
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
- title3="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).",
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()