dim fix

app.py

@@ -228,6 +228,11 @@ def process_fmri(file_obj):
         
         if data.ndim != 4:
             return f"error: expected 4D data, got {data.ndim}D", None
+        
+        if data.ndim == 3:
+            data = data[None,...]  # add time dim [H,W,D] -> [1,H,W,D]
+        elif data.ndim != 4:
+            return f"error: expected 3D/4D data, got {data.ndim}D", None
             
         data = preprocess_volume(data)
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')