sabrinabenas commited on
Commit
b427ece
β€’
1 Parent(s): e4c380d
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@
3
  # Built from https://huggingface.co/spaces/Neslihan/megadetector_dlcmodels/blob/main/app.py
4
 
5
 
 
6
  import gradio as gr
7
 
8
  import torch
@@ -85,7 +86,7 @@ def draw_keypoints_on_image(image,
85
  round_outline = [round(num*255) for num in list(cmap2(alpha[i]))[:3]]
86
  draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),
87
  (keypoint_x + marker_size, keypoint_y + marker_size)],
88
- fill=tuple(round_fill), outline= tuple(round_outline))
89
 
90
  # add string labels around keypoints
91
  if flag_show_str_labels:
 
3
  # Built from https://huggingface.co/spaces/Neslihan/megadetector_dlcmodels/blob/main/app.py
4
 
5
 
6
+ from tkinter import W
7
  import gradio as gr
8
 
9
  import torch
 
86
  round_outline = [round(num*255) for num in list(cmap2(alpha[i]))[:3]]
87
  draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),
88
  (keypoint_x + marker_size, keypoint_y + marker_size)],
89
+ fill=tuple(round_fill), outline= tuple(round_outline), width=3)
90
 
91
  # add string labels around keypoints
92
  if flag_show_str_labels: