gpu

app.py

@@ -93,6 +93,7 @@ model_index = 0
 # ヒートマップの生成関数
 @spaces.GPU
 def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
+    global model_index, mean_vector_list
     if type(uploaded_image) == str:
         uploaded_image = Image.open(uploaded_image)
     if type(source_num) == str:
@@ -152,6 +153,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
+    
+    print("setup done.")
 
 with gr.Blocks() as demo:
     # title