Update app.py

app.py

@@ -58,10 +58,10 @@ def compute_motion_map(spike, tau_f, tau_s, U_c, theta, sampling_rate=40000):
     # 取中心时间步的运动掩码
     central_index = T // 2
     motion_map = motion_masks[central_index]
-    print("ret")
+    print(motion_map.min(), motion_map.max())
     # 转换为图像格式 (0-255)
     motion_map = (motion_map * 255).astype(np.uint8)
-    print("returning")
+    
     return motion_map
 
 # Gradio 接口
@@ -114,7 +114,7 @@ with gr.Blocks() as demo:
         spike = load_vidar_dat(dat_path, width=400, height=250)
         # 计算运动图
         motion_map = compute_motion_map(spike, tau_f, tau_s, U_c, theta)
-        print("rett")
+        
         # 转换为 PIL 图像以供 Gradio 使用
         return Image.fromarray(motion_map)