並び替え

app.py

@@ -88,10 +88,13 @@ for model_info in models_info:
 
 x = load_data()
 test_imgs, points = load_keypoints()
+model = None
 
 # ヒートマップの生成関数
 @spaces.GPU
 def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
+    global model, mean_vector_list
+    
     if type(uploaded_image) == str:
         uploaded_image = Image.open(uploaded_image)
     if type(source_num) == str:
@@ -206,6 +209,5 @@ if __name__ == "__main__":
         
     setup(models_info[0])
 
-    # JavaScriptコードをロード
     demo.launch()