install torchvision with torch

app.py

@@ -1,15 +1,17 @@
 # PyTorch 2.8 (temporary hack)
 import os
-os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
 
+os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9 torchvision" spaces')
+import torchvision.transforms.functional as TF
 # Actual demo code
 import gradio as gr
 import numpy as np
 import spaces
+
 import torch
 import random
 from PIL import Image
-import torchvision.transforms.functional as TF
+
 from pipeline import GenSIRR
 from diffusers.utils import load_image
 import torch.nn.functional as F