Update test_ai_integration_http.py

test_ai_integration_http.py

@@ -12,7 +12,7 @@ from typing import Any, Optional
 import torch
 from torch import nn
 import torch.nn.functional as F
-from torch.utils._python_dispatch import TorchFunctionMode
+from torch.overrides import TorchFunctionMode
 from PIL import Image
 from transformers import (
     AutoProcessor,