gpu

app.py

@@ -83,12 +83,12 @@ for model_info in models_info:
     model_name = model_info["name"]
     feature_dim = model_info["feature_dim"]
     model_path = f"checkpoints/{model_name}"
-    model = load_model(model_path, feature_dim)
-    models.append(model)
+    models.append(load_model(model_path, feature_dim))
 
 x = load_data()
 test_imgs, points = load_keypoints()
-model = None
+mean_vector_list = []
+model_index = 0
 
 # ヒートマップの生成関数
 @spaces.GPU
@@ -141,15 +141,14 @@ def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
 
 @spaces.GPU
 def setup(model_info, input_image=None):
-    global model, mean_vector_list
+    global model_index, mean_vector_list
     # str -> dictに変換
     if type(model_info) == str:
         model_info = eval(model_info)
-        
-    index = models_info.index(model_info)
-    model = models[index]
 
-    feature_map, _ = model(test_imgs)
+    model_index = models_info.index(model_info)
+
+    feature_map, _ = models[model_index](test_imgs)
     mean_vector_list = utils.get_mean_vector(feature_map, points)
     
     if input_image is not None: