solve problem with input secont try

app.py

@@ -5,19 +5,21 @@ from PIL import Image
 import cv2
 
 # Загрузка модели
-midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+midas = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device)
 midas.eval()
 
-midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms")
+midas_transforms = torch.hub.load("intel-isl/MiDaS", "transforms").to(device)
 transform = midas_transforms.small_transform
 
+
 def predict_depth(image):
     img = np.array(image)
     img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    input_tensor = transform(img_rgb)
+    input_tensor = transform(img_rgb).to(device)
 
     with torch.no_grad():
-        prediction = midas(input_tensor.unsqueeze(0))
+        prediction = midas(input_tensor)
         prediction = torch.nn.functional.interpolate(
             prediction.unsqueeze(1),
             size=img_rgb.shape[:2],