Update app.py

app.py

@@ -5,7 +5,6 @@ import onnx
 import onnxruntime as rt
 from numpy.random import RandomState
 from skimage import transform
-import json
 
 
 def get_inter(r1, r2):
@@ -181,10 +180,10 @@ class Model:
                 if detail:
                     print(msg + f'body num error: {body_num}')
                 continue
-            bounds = (int(min(waifu_rect[0], hr[0], br[0])),
-                      int(min(waifu_rect[1], hr[1], br[1])),
-                      int(max(waifu_rect[2], hr[2], br[2])),
-                      int(max(waifu_rect[3], hr[3], br[3])))
+            bounds = (min(waifu_rect[0], hr[0], br[0]),
+                      min(waifu_rect[1], hr[1], br[1]),
+                      max(waifu_rect[2], hr[2], br[2]),
+                      max(waifu_rect[3], hr[3], br[3]))
             if (bounds[2] - bounds[0]) / (bounds[3] - bounds[1]) > 0.7:
                 if detail:
                     print(msg + "ratio out of limit")
@@ -258,7 +257,8 @@ if __name__ == '__main__':
 
     app = gr.Blocks()
     with app:
-        gr.Markdown("# full-body anime\n\n"
+        gr.Markdown("# full-body anime GAN\n\n"
+                    "![visitor badge](https://visitor-badge.glitch.me/badge?page_id=skytnt.full-body-anime-gan)\n\n"
                     "the model is not well, just use for fun.")
         with gr.Tabs():
             with gr.TabItem("generate image"):
@@ -270,7 +270,7 @@ if __name__ == '__main__':
                             gen_input2 = gr.Number(value=1, label="seed ( int between -2^31 and 2^31 - 1 )")
                         gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.6, label="truncation psi")
                         with gr.Group():
-                            gen_submit = gr.Button("Generate")
+                            gen_submit = gr.Button("Generate", variant="primary")
                     with gr.Column():
                         gen_output1 = gr.Image(label="output image")
                         select_img_input_w1 = gr.Variable()
@@ -282,7 +282,7 @@ if __name__ == '__main__':
                         gr.Markdown("you'd better upload a standing full-body image")
                         encode_img_input = gr.Image(label="input image")
                         with gr.Group():
-                            encode_img_submit = gr.Button("Run")
+                            encode_img_submit = gr.Button("Run", variant="primary")
                     with gr.Column():
                         encode_img_output1 = gr.Textbox(label="output message")
                         with gr.Row():
@@ -302,7 +302,7 @@ if __name__ == '__main__':
                                                                 choices=["current generated image",
                                                                          "current encoded image"], type="index")
                                 with gr.Group():
-                                    select_img1_button = gr.Button("select")
+                                    select_img1_button = gr.Button("select", variant="primary")
                                 select_img1_output_img = gr.Image(label="selected image 1")
                                 select_img1_output_w = gr.Variable()
                             with gr.Column():
@@ -311,15 +311,14 @@ if __name__ == '__main__':
                                                                 choices=["current generated image",
                                                                          "current encoded image"], type="index")
                                 with gr.Group():
-                                    select_img2_button = gr.Button("select")
+                                    select_img2_button = gr.Button("select", variant="primary")
                                 select_img2_output_img = gr.Image(label="selected image 2")
                                 select_img2_output_w = gr.Variable()
                         generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=15)
                         with gr.Group():
-                            generate_video_button = gr.Button("Generate")
+                            generate_video_button = gr.Button("Generate", variant="primary")
                     with gr.Column():
                         generate_video_output = gr.Video(label="output video")
-
         gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3],
                          [gen_output1, select_img_input_w1, select_img_input_img1])
         encode_img_submit.click(encode_img_fn, [encode_img_input],
@@ -335,5 +334,4 @@ if __name__ == '__main__':
                                  [select_img2_output_img, select_img2_output_w])
         generate_video_button.click(gen_video_fn, [select_img1_output_w, select_img2_output_w, generate_video_frame],
                                     [generate_video_output])
-
     app.launch()