Update app.py

app.py

@@ -42,7 +42,7 @@ def inference(img):
     image = image_transform(image)
     
     # The model expects inputs of shape: B x C x T x H x W
-    image = image[None, :, None, ...]
+    image = image.unsqueeze(0)
     
     prediction = model(image)
     prediction = F.softmax(prediction, dim=1)