並び替え

app.py

@@ -77,7 +77,7 @@ models_info = [
     {"name": "autoencoder-epoch=29-train_loss=1.01.ckpt", "feature_dim": 64},
     {"name": "autoencoder-epoch=09-train_loss=1.00.ckpt", "feature_dim": 64},  
     {"name": "ae_model_tf_2024-03-05_00-35-21.pth", "feature_dim": 32},
-    ]
+]
 models = []
 for model_info in models_info:
     model_name = model_info["name"]
@@ -155,6 +155,8 @@ 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
@@ -205,9 +207,5 @@ with gr.Blocks() as demo:
         inputs=[input_image],
     )
 
-if __name__ == "__main__":
-        
-    setup(models_info[0])
-
     demo.launch()