fix final output 300,300

app.py

@@ -130,9 +130,13 @@ def generate_flow_intermediates_streaming(label):
         if i in [10,20,30,40,48,49]:
             #images.append((x + 1) / 2.0)
             step_idx = {10: 1, 20: 2, 30: 3, 40: 4, 48: 5, 49:12}[i]
-            outputs[step_idx] = resize(((x + 1) / 2.0)[0, 0].clamp(0, 1).cpu().numpy())
+            if i==49:
+                outputs[step_idx] = resize(((x + 1) / 2.0)[0, 0].clamp(0, 1).cpu().numpy(),(300,300))
+            else:
+                outputs[step_idx] = resize(((x + 1) / 2.0)[0, 0].clamp(0, 1).cpu().numpy())
             yield tuple(outputs)
 
+
             # Compute velocity magnitude and convert to numpy for visualization
         if i in [0,11,21,31,41,49]:
             v_mag = dt*v[0, 0].abs().clamp(0, 3).cpu().numpy()  # Clamp to max value for better contrast
@@ -140,10 +144,7 @@ def generate_flow_intermediates_streaming(label):
             vel_colored = plt.get_cmap("coolwarm")(v_mag)[:, :, :3]  # (H,W,3)
             vel_colored = (vel_colored * 255).astype(np.uint8)
             step_idx = {0: 6, 11: 7, 21: 8, 31: 9, 41: 10, 49:11}[i]
-            if i==49:
-                outputs[step_idx] = resize(vel_colored, (400, 400))
-            else:
-                outputs[step_idx] = resize(vel_colored)
+            outputs[step_idx] = resize(vel_colored)
             yield tuple(outputs)