並び替え

app.py

@@ -155,8 +155,6 @@ def setup(model_info, input_image=None):
     if input_image is not None:
         fig = get_heatmaps(0, image_size // 2, image_size // 2, input_image)
         return fig
-        
-setup(models_info[0])
 
 with gr.Blocks() as demo:
     # title
@@ -207,5 +205,7 @@ with gr.Blocks() as demo:
         inputs=[input_image],
     )
 
+    setup(models_info[0])
+
     demo.launch()