Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from PIL import Image
 from utils_spike import load_vidar_dat, STPFilter  # 假设您有这个函数
 import os
+form tqdm import tqdm, trange
 
 # 设置设备
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -49,7 +50,7 @@ def compute_motion_map(spike, tau_f, tau_s, U_c, theta, sampling_rate=40000):
 
     # 逐时间步更新并获取运动掩码
     motion_masks = []
-    for t in range(T):
+    for t in trange(T):
         cur_spikes = spike[t]
         stp_filter.update_dynamics(t, cur_spikes)
         motion_masks.append(stp_filter.filter_spk.cpu().numpy())